diff options
-rw-r--r-- | nova/test.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/nova/test.py b/nova/test.py index d1c1ad20e..d759aef60 100644 --- a/nova/test.py +++ b/nova/test.py @@ -278,20 +278,20 @@ class TestCase(unittest.TestCase): else: self.assertEqual(sub_value, super_value) - def assertIn(self, a, b): + def assertIn(self, a, b, *args, **kwargs): """Python < v2.7 compatibility. Assert 'a' in 'b'""" try: f = super(TestCase, self).assertIn except AttributeError: - self.assertTrue(a in b) + self.assertTrue(a in b, *args, **kwargs) else: - f(a, b) + f(a, b, *args, **kwargs) - def assertNotIn(self, a, b): + def assertNotIn(self, a, b, *args, **kwargs): """Python < v2.7 compatibility. Assert 'a' NOT in 'b'""" try: f = super(TestCase, self).assertNotIn except AttributeError: - self.assertFalse(a in b) + self.assertFalse(a in b, *args, **kwargs) else: - f(a, b) + f(a, b, *args, **kwargs) |