|
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