|
1 | 1 | import datetime |
2 | 2 | import unittest |
3 | 3 |
|
4 | | -from sys import version_info |
5 | | - |
6 | 4 | import odml.dtypes as typ |
7 | 5 |
|
8 | 6 |
|
9 | 7 | class TestTypes(unittest.TestCase): |
10 | 8 |
|
11 | | - def assert_local_regexp(self, text, regular_expression): |
12 | | - """ |
13 | | - Python 2 is dead and assertRegexpMatches is deprecated and |
14 | | - will be removed, but keep compatibility until py 2 support is |
15 | | - fully dropped. |
16 | | - """ |
17 | | - |
18 | | - if version_info.major < 3: |
19 | | - self.assertRegexpMatches(text, regular_expression) |
20 | | - else: |
21 | | - self.assertRegex(text, regular_expression) |
22 | | - |
23 | 9 | def test_valid_type(self): |
24 | 10 | # Test None |
25 | 11 | self.assertTrue(typ.valid_type(None)) |
@@ -47,8 +33,8 @@ def test_date(self): |
47 | 33 | self.assertIsInstance(typ.date_get(""), datetime.date) |
48 | 34 |
|
49 | 35 | reg = "^[0-9]{4}-(0[1-9]|1[0-2])-([0-2][0-9]|3[0-1])$" |
50 | | - self.assert_local_regexp(typ.date_get(None).strftime(typ.FORMAT_DATE), reg) |
51 | | - self.assert_local_regexp(typ.date_get("").strftime(typ.FORMAT_DATE), reg) |
| 36 | + self.assertRegex(typ.date_get(None).strftime(typ.FORMAT_DATE), reg) |
| 37 | + self.assertRegex(typ.date_get("").strftime(typ.FORMAT_DATE), reg) |
52 | 38 |
|
53 | 39 | date = datetime.date(2011, 12, 1) |
54 | 40 | date_string = '2011-12-01' |
@@ -79,8 +65,8 @@ def test_time(self): |
79 | 65 | self.assertIsInstance(typ.time_get(""), datetime.time) |
80 | 66 |
|
81 | 67 | reg = "^[0-5][0-9]:[0-5][0-9]:[0-5][0-9]$" |
82 | | - self.assert_local_regexp(typ.time_get(None).strftime(typ.FORMAT_TIME), reg) |
83 | | - self.assert_local_regexp(typ.time_get("").strftime(typ.FORMAT_TIME), reg) |
| 68 | + self.assertRegex(typ.time_get(None).strftime(typ.FORMAT_TIME), reg) |
| 69 | + self.assertRegex(typ.time_get("").strftime(typ.FORMAT_TIME), reg) |
84 | 70 |
|
85 | 71 | time = datetime.time(12, 34, 56) |
86 | 72 | time_string = '12:34:56' |
@@ -111,8 +97,8 @@ def test_datetime(self): |
111 | 97 | self.assertIsInstance(typ.datetime_get(""), datetime.datetime) |
112 | 98 |
|
113 | 99 | reg = "^[0-9]{4}-(0[1-9]|1[0-2])-([0-2][0-9]|3[0-1]) [0-5][0-9]:[0-5][0-9]:[0-5][0-9]$" |
114 | | - self.assert_local_regexp(typ.datetime_get(None).strftime(typ.FORMAT_DATETIME), reg) |
115 | | - self.assert_local_regexp(typ.datetime_get("").strftime(typ.FORMAT_DATETIME), reg) |
| 100 | + self.assertRegex(typ.datetime_get(None).strftime(typ.FORMAT_DATETIME), reg) |
| 101 | + self.assertRegex(typ.datetime_get("").strftime(typ.FORMAT_DATETIME), reg) |
116 | 102 |
|
117 | 103 | date = datetime.datetime(2011, 12, 1, 12, 34, 56) |
118 | 104 | date_string = '2011-12-01 12:34:56' |
|
0 commit comments