1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
import datetime
import pickle
import unittest
import pytest
from geopy.timezone import Timezone, from_fixed_gmt_offset, from_timezone_name
try:
import pytz
import pytz.tzinfo
pytz_available = True
except ImportError:
pytz_available = False
@pytest.mark.skipif("not pytz_available")
class TimezoneTestCase(unittest.TestCase):
timezone_gmt_offset_hours = 3.0
timezone_name = "Europe/Moscow" # a DST-less timezone
def test_create_from_timezone_name(self):
raw = dict(foo="bar")
tz = from_timezone_name(self.timezone_name, raw)
self.assertEqual(tz.raw['foo'], 'bar')
self.assertIsInstance(tz.pytz_timezone, pytz.tzinfo.BaseTzInfo)
self.assertIsInstance(tz.pytz_timezone, datetime.tzinfo)
def test_create_from_fixed_gmt_offset(self):
raw = dict(foo="bar")
tz = from_fixed_gmt_offset(self.timezone_gmt_offset_hours, raw)
self.assertEqual(tz.raw['foo'], 'bar')
# pytz.FixedOffset is not an instanse of pytz.tzinfo.BaseTzInfo.
self.assertIsInstance(tz.pytz_timezone, datetime.tzinfo)
olson_tz = pytz.timezone(self.timezone_name)
dt = datetime.datetime.utcnow()
self.assertEqual(tz.pytz_timezone.utcoffset(dt), olson_tz.utcoffset(dt))
def test_create_from_pytz_timezone(self):
pytz_timezone = pytz.timezone(self.timezone_name)
tz = Timezone(pytz_timezone, {})
self.assertIs(tz.pytz_timezone, pytz_timezone)
def test_string(self):
raw = dict(foo="bar")
tz = from_timezone_name(self.timezone_name, raw)
self.assertEqual(str(tz), self.timezone_name)
def test_repr(self):
raw = dict(foo="bar")
pytz_timezone = pytz.timezone(self.timezone_name)
tz = Timezone(pytz_timezone, raw)
self.assertEqual(repr(tz), "Timezone(%s)" % repr(pytz_timezone))
def test_eq(self):
tz = pytz.timezone("Europe/Paris")
raw1 = dict(a=1)
raw2 = dict(a=1)
self.assertEqual(Timezone(tz, raw1), Timezone(tz, raw2))
def test_ne(self):
tz1 = pytz.timezone("Europe/Paris")
tz2 = pytz.timezone("Europe/Prague")
raw = {}
self.assertNotEqual(Timezone(tz1, raw), Timezone(tz2, raw))
def test_picklable(self):
raw = dict(foo="bar")
tz = from_timezone_name(self.timezone_name, raw)
# https://docs.python.org/2/library/pickle.html#data-stream-format
for protocol in (0, 1, 2, -1):
pickled = pickle.dumps(tz, protocol=protocol)
tz_unp = pickle.loads(pickled)
self.assertEqual(tz, tz_unp)
def test_with_unpicklable_raw(self):
some_class = type('some_class', (object,), {})
raw_unpicklable = dict(missing=some_class())
del some_class
tz_unpicklable = from_timezone_name(self.timezone_name, raw_unpicklable)
for protocol in (0, 1, 2, -1):
with self.assertRaises((AttributeError, pickle.PicklingError)):
pickle.dumps(tz_unpicklable, protocol=protocol)
|
