Skip to content

Commit 78c4639

Browse files
committed
Backport more missing unittest functions to py2.6
1 parent 5c6eae3 commit 78c4639

File tree

3 files changed

+20
-4
lines changed

3 files changed

+20
-4
lines changed

tests/accessories.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def norm_file(fname):
1515

1616

1717
class TestCaseBackport(unittest.TestCase):
18-
"""Backport assertRaises and assertIn for python2.5 and 2.6."""
18+
"""Backport python2.7 TestCase methods (in leiu of unittest2, or better, pytest)."""
1919
if sys.version_info < (2, 7):
2020

2121
def assertIn(self, member, container, msg=None):
@@ -25,6 +25,21 @@ def assertIn(self, member, container, msg=None):
2525
safe_repr(container))
2626
self.fail(self._formatMessage(msg, standardMsg))
2727

28+
def assertIs(self, expr1, expr2, msg=None):
29+
"""Just like self.assertTrue(a is b), but with a nicer default message."""
30+
if expr1 is not expr2:
31+
standardMsg = '%s is not %s' % (safe_repr(expr1),
32+
safe_repr(expr2))
33+
self.fail(self._formatMessage(msg, standardMsg))
34+
35+
36+
def assertIsInstance(self, obj, cls, msg=None):
37+
"""Same as self.assertTrue(isinstance(obj, cls)), with a nicer
38+
default message."""
39+
if not isinstance(obj, cls):
40+
standardMsg = '%s is not an instance of %r' % (safe_repr(obj), cls)
41+
self.fail(self._formatMessage(msg, standardMsg))
42+
2843
@contextlib.contextmanager
2944
def assertRaises(self, excClass):
3045
try:

tests/test_onimport.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from accessories import TestCaseBackport
55

66

7-
class SqliteDict_cPickleImportTest(unittest.TestCase):
7+
class SqliteDict_cPickleImportTest(TestCaseBackport):
88
"""Verify fallback to 'pickle' module when 'cPickle' is not found."""
99
def setUp(self):
1010
self.orig_meta_path = sys.meta_path
@@ -15,7 +15,6 @@ def __init__(self, *args):
1515
self.module_names = args
1616

1717
def find_module(self, fullname, path=None):
18-
print(fullname)
1918
if fullname in self.module_names:
2019
return self
2120
return None

tests/test_temp_db.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
11
import unittest
22
import sqlitedict
33

4+
from accessories import TestCaseBackport
5+
46
from sys import version_info
57
_major_version=version_info[0]
68

7-
class TempSqliteDictTest(unittest.TestCase):
9+
class TempSqliteDictTest(TestCaseBackport):
810

911
def setUp(self):
1012
self.d = sqlitedict.SqliteDict()

0 commit comments

Comments
 (0)