Skip to content

Commit eab2504

Browse files
committed
Test fixes for Python 2
1 parent 807eb9c commit eab2504

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/models_test.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,14 @@
77
sys.path.append('..')
88

99
import geoip2.models
10-
import unittest
10+
11+
if sys.version_info[:2] == (2, 6):
12+
import unittest2 as unittest
13+
else:
14+
import unittest
15+
16+
if sys.version_info[0] == 2:
17+
unittest.TestCase.assertRaisesRegex = unittest.TestCase.assertRaisesRegexp
1118

1219

1320
class TestModels(unittest.TestCase):

0 commit comments

Comments
 (0)