openstack/nova
Code Issues Proposed changes

support the extra optional arguments for msg to assertIn and assertNotIn

Browse Source
This commit is contained in:
Chris Behrens 2011-08-29 01:13:08 -07:00
parent 599467124e
commit 8bfa5e23e9
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nova/test.py
View File

@ -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)