From 52e91d5f9c6e2ba659965d5043a247201466dffb Mon Sep 17 00:00:00 2001 From: Valentin Lorentz Date: Sat, 5 Jan 2013 21:16:30 +0100 Subject: [PATCH] src/test.py: Implement assertIn for old Python versions. --- src/test.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/test.py b/src/test.py index ef12fce70..56d88f9f9 100644 --- a/src/test.py +++ b/src/test.py @@ -115,6 +115,20 @@ class SupyTestCase(unittest.TestCase): irc._reallyDie() if sys.version_info < (2, 7, 0): + def assertIn(self, member, container, msg=None): + """Just like self.assertTrue(a in b), but with a nicer default message.""" + if member not in container: + standardMsg = '%s not found in %s' % (safe_repr(member), + safe_repr(container)) + self.fail(self._formatMessage(msg, standardMsg)) + + def assertNotIn(self, member, container, msg=None): + """Just like self.assertTrue(a not in b), but with a nicer default message.""" + if member in container: + standardMsg = '%s unexpectedly found in %s' % (safe_repr(member), + safe_repr(container)) + self.fail(self._formatMessage(msg, standardMsg)) + def assertIs(self, expr1, expr2, msg=None): """Just like self.assertTrue(a is b), but with a nicer default message.""" if expr1 is not expr2: