diff --git a/headphones/unittestcompat.py b/headphones/unittestcompat.py index d77c6a16..f213295e 100644 --- a/headphones/unittestcompat.py +++ b/headphones/unittestcompat.py @@ -34,6 +34,12 @@ class TestCase(TC): def assertIn(self, *args, **kw): return super(TestCase, self).assertIn(*args, **kw) + def assertIsNone(self, val, msg=None): + if not _dummy: + return super(TestCase, self).assertIsNone(val, msg) + tst = val is None + return super(TestCase, self).assertTrue(tst, msg) + def assertIsNotNone(self, val, msg=None): if not _dummy: return super(TestCase, self).assertIsNotNone(val, msg)