|
7 | 7 | """ |
8 | 8 |
|
9 | 9 | # stdlib |
10 | | -import datetime |
11 | 10 | import re |
12 | | -from datetime import date |
| 11 | +from datetime import date, datetime, timedelta |
13 | 12 |
|
14 | 13 | # 3rd party |
15 | 14 | import pytest |
|
19 | 18 | from domdf_python_tools.dates import calc_easter |
20 | 19 | from domdf_python_tools.testing import count |
21 | 20 |
|
22 | | -# TODO: travis matrix to test without pytz installed |
23 | 21 | # TODO: test get_timezone |
24 | 22 |
|
25 | 23 | try: |
26 | 24 | # 3rd party |
27 | 25 | import pytz |
28 | 26 |
|
29 | | - test_date = datetime.datetime(1996, 10, 13, 2, 20).replace(tzinfo=pytz.utc) |
30 | | - today = datetime.datetime.utcnow().replace(tzinfo=pytz.utc) # make sure UTC |
| 27 | + test_date = datetime(1996, 10, 13, 2, 20).replace(tzinfo=pytz.utc) |
| 28 | + today = datetime.utcnow().replace(tzinfo=pytz.utc) # make sure UTC |
31 | 29 |
|
32 | 30 | def test_utc_offset(): |
33 | 31 | # Check that the correct UTC offsets are given for common timezones |
34 | | - assert dates.get_utc_offset("US/Pacific", test_date) == datetime.timedelta(-1, 61200) |
35 | | - assert dates.get_utc_offset("Europe/London", test_date) == datetime.timedelta(0, 3600) |
36 | | - assert dates.get_utc_offset("Africa/Algiers", test_date) == datetime.timedelta(0, 3600) |
| 32 | + assert dates.get_utc_offset("US/Pacific", test_date) == timedelta(-1, 61200) |
| 33 | + assert dates.get_utc_offset("Europe/London", test_date) == timedelta(0, 3600) |
| 34 | + assert dates.get_utc_offset("Africa/Algiers", test_date) == timedelta(0, 3600) |
37 | 35 | # TODO: Finish |
38 | 36 |
|
39 | 37 | # Check that the correct UTC offsets are given for common timezones for today |
40 | 38 | assert dates.get_utc_offset("US/Pacific", today) in { |
41 | | - datetime.timedelta(-1, 57600), |
42 | | - datetime.timedelta(-1, 61200), |
| 39 | + timedelta(-1, 57600), |
| 40 | + timedelta(-1, 61200), |
43 | 41 | } |
44 | 42 | assert dates.get_utc_offset("Europe/London", today) in { |
45 | | - datetime.timedelta(0, 3600), # BST |
46 | | - datetime.timedelta(0, 0), |
| 43 | + timedelta(0, 3600), # BST |
| 44 | + timedelta(0, 0), |
47 | 45 | } |
48 | | - assert dates.get_utc_offset("Africa/Algiers", today) == datetime.timedelta(0, 3600) |
| 46 | + assert dates.get_utc_offset("Africa/Algiers", today) == timedelta(0, 3600) |
49 | 47 |
|
50 | 48 | # Check that the correct UTC offsets are given for common timezones when ``date`` is not given |
51 | 49 | assert dates.get_utc_offset("US/Pacific") in { |
52 | | - datetime.timedelta(-1, 57600), |
53 | | - datetime.timedelta(-1, 61200), |
| 50 | + timedelta(-1, 57600), |
| 51 | + timedelta(-1, 61200), |
54 | 52 | } |
55 | 53 | assert dates.get_utc_offset("Europe/London") in { |
56 | | - datetime.timedelta(0, 3600), # BST |
57 | | - datetime.timedelta(0, 0), |
| 54 | + timedelta(0, 3600), # BST |
| 55 | + timedelta(0, 0), |
58 | 56 | } |
59 | | - assert dates.get_utc_offset("Africa/Algiers") == datetime.timedelta(0, 3600) |
| 57 | + assert dates.get_utc_offset("Africa/Algiers") == timedelta(0, 3600) |
60 | 58 |
|
61 | 59 | def test_converting_timezone(): |
62 | 60 | # No matter what timezone we convert to the timestamp should be the same |
|
0 commit comments