From b5fb769bff27ae8f734634083dc3090816e5435a Mon Sep 17 00:00:00 2001 From: maxkoryukov Date: Wed, 3 Feb 2016 06:59:10 +0500 Subject: [PATCH] improved test compatability --- headphones/unittestcompat.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/headphones/unittestcompat.py b/headphones/unittestcompat.py index 87427bd1..0cbb901a 100644 --- a/headphones/unittestcompat.py +++ b/headphones/unittestcompat.py @@ -24,5 +24,9 @@ class TestCase(TC): return super(TestCase, self).assertIsInstance(*args, **kw) @_d - def assertIsIn(self, *args, **kw): - return super(TestCase, self).assertIsIn(*args, **kw) + def assertIn(self, *args, **kw): + return super(TestCase, self).assertIn(*args, **kw) + + @_d + def assertIsNotNone(self, *args, **kw): + return super(TestCase, self).assertIsNotNone(*args, **kw)