neingeist/check_kernel
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Fix patching up unittest.TestCase to support assertGreater for ancient Python

Browse source
This commit is contained in:
Mike Gerber 2015-05-25 06:11:51 +02:00
parent 52de7c2da7
commit fb36038bee
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
check_kernel_test.py
View file

@ -18,11 +18,11 @@ def patch_object(obj, function, return_value=None):
obj.__dict__[function] = our_function
# Patch up unittest
if not hasattr(unittest, 'assertGreater'):
unittest.assertGreater = lambda self, a, b: self.assertTrue(a > b)
if not hasattr(unittest, 'assertLess'):
unittest.assertLess = lambda self, a, b: self.assertTrue(a < b)
# Patch up unittest.TestCase
if not hasattr(unittest.TestCase, 'assertGreater'):
unittest.TestCase.assertGreater = lambda self, a, b: self.assertTrue(a > b)
if not hasattr(unittest.TestCase, 'assertLess'):
unittest.TestCase.assertLess = lambda self, a, b: self.assertTrue(a < b)
class CleanKernelVersionTestCase(unittest.TestCase):