From fb36038beeb8416ffd948d3c99d846c552ec52d9 Mon Sep 17 00:00:00 2001 From: Mike Gerber Date: Mon, 25 May 2015 06:11:51 +0200 Subject: [PATCH] Fix patching up unittest.TestCase to support assertGreater for ancient Python --- check_kernel_test.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/check_kernel_test.py b/check_kernel_test.py index 91945a7..0596ca4 100644 --- a/check_kernel_test.py +++ b/check_kernel_test.py @@ -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):