From 5034e7b4f6a91a838063d75cb786c143c9c08d2c Mon Sep 17 00:00:00 2001 From: Matt Wright Date: Wed, 20 Jun 2012 16:21:35 -0400 Subject: [PATCH] Add test methods that are missing for Python versions < 2.7 --- tests/__init__.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/__init__.py b/tests/__init__.py index 63cdbbc..7af88b9 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -41,3 +41,21 @@ class SecurityTest(TestCase): def assertIsHomePage(self, data): self.assertIn('Home Page', data) + + def assertIn(self, member, container, msg=None): + if hasattr(TestCase, 'assertIn'): + return TestCase.assertIn(self, member, container, msg) + + return self.assertTrue(member in container) + + def assertNotIn(self, member, container, msg=None): + if hasattr(TestCase, 'assertNotIn'): + return TestCase.assertNotIn(self, member, container, msg) + + return self.assertFalse(member in container) + + def assertIsNotNone(self, obj, msg=None): + if hasattr(TestCase, 'assertIsNotNone'): + return TestCase.assertIsNotNone(self, obj, msg) + + return self.assertTrue(obj is not None)