def test_comparison():
now = datetime.datetime.now()
later = now.replace(microsecond=0) + datetime.timedelta(hours=1)
v1 = temporenc.unpackb(temporenc.packb(now))
v2 = temporenc.unpackb(temporenc.packb(now))
v3 = temporenc.unpackb(temporenc.packb(later))
# Same
assert v1 == v2
assert v1 != v3
assert v1 >= v2
assert v1 >= v2
assert not (v1 > v2)
assert not (v1 < v2)
# Different
assert v3 > v1
assert v1 < v3
assert v3 >= v1
assert v1 <= v3
# Equality tests against other types: not equal
bogus = 'junk'
assert not (v1 == bogus)
assert v1 != bogus
# Comparison against other types:
# * fail on Python 3 (unorderable types)
# * use fallback comparison on Python 2.
for op in (operator.gt, operator.lt, operator.ge, operator.le):
if PY2:
op(v1, bogus) # should not raise
else:
with pytest.raises(TypeError):
op(v1, bogus) # should raise
def test_comparison():
now = datetime.datetime.now()
later = now.replace(microsecond=0) + datetime.timedelta(hours=1)
v1 = temporenc.unpackb(temporenc.packb(now))
v2 = temporenc.unpackb(temporenc.packb(now))
v3 = temporenc.unpackb(temporenc.packb(later))
# Same
assert v1 == v2
assert v1 != v3
assert v1 >= v2
assert v1 >= v2
assert not (v1 > v2)
assert not (v1 < v2)
# Different
assert v3 > v1
assert v1 < v3
assert v3 >= v1
assert v1 <= v3
# Equality tests against other types: not equal
bogus = 'junk'
assert not (v1 == bogus)
assert v1 != bogus
# Comparison against other types:
# * fail on Python 3 (unorderable types)
# * use fallback comparison on Python 2.
for op in (operator.gt, operator.lt, operator.ge, operator.le):
if PY2:
op(v1, bogus) # should not raise
else:
with pytest.raises(TypeError):
op(v1, bogus) # should raise
def test_native_unpacking():
value = temporenc.unpackb(temporenc.packb(
year=1983, month=1, day=15))
assert value.date() == datetime.date(1983, 1, 15)
value = temporenc.unpackb(temporenc.packb(
year=1983, month=1, day=15,
hour=1, minute=2, second=3, microsecond=456))
assert value.datetime() == datetime.datetime(1983, 1, 15, 1, 2, 3, 456)
value = temporenc.unpackb(temporenc.packb(
year=1983, month=1, day=15, # will be ignored
hour=1, minute=2, second=3, microsecond=456))
assert value.time() == datetime.time(1, 2, 3, 456)
value = temporenc.unpackb(temporenc.packb(year=1234))
with pytest.raises(ValueError):
value.date()
assert value.date(strict=False).year == 1234
assert value.datetime(strict=False).year == 1234
value = temporenc.unpackb(temporenc.packb(hour=14))
with pytest.raises(ValueError):
value.time()
assert value.time(strict=False).hour == 14
assert value.datetime(strict=False).hour == 14
def test_native_time_zone():
# Python < 3.2 doesn't have concrete tzinfo implementations. This
# test uses the internal helper class instead to avoid depending on
# newer Python versions (or on pytz).
from temporenc.temporenc import FixedOffset
dutch_winter = FixedOffset(60) # UTC +01:00
zero_delta = datetime.timedelta(0)
hour_delta = datetime.timedelta(minutes=60)
expected_name = "UTC+01:00"
assert dutch_winter.tzname(None) == expected_name
assert expected_name in str(dutch_winter)
assert expected_name in repr(dutch_winter)
assert dutch_winter.dst(None) == zero_delta
# DTZ
actual = temporenc.packb(datetime.datetime(1983,
1,
15,
18,
25,
12,
0,
tzinfo=dutch_winter),
type='DTZ')
expected = from_hex('cf 7e 0e 93 26 44')
assert actual == expected
moment = temporenc.unpackb(expected)
assert moment.hour == 18
assert moment.tz_offset == 60
dt = moment.datetime()
assert dt.hour == 18
assert dt.utcoffset() == hour_delta
# DTSZ (microsecond, since native types have that precision)
actual = temporenc.packb(datetime.datetime(1983,
1,
15,
18,
25,
12,
123456,
tzinfo=dutch_winter),
type='DTSZ')
dtsz_us = from_hex('eb df 83 a4 c9 83 c4 81 10')
assert actual == dtsz_us
moment = temporenc.unpackb(expected)
assert moment.datetime().hour == 18
# Time only with time zone
moment = temporenc.unpackb(
temporenc.packb(datetime.time(0, 30, 0, 123456, tzinfo=dutch_winter),
type='DTSZ'))
assert moment.tz_offset == 60
for obj in [moment.time(), moment.datetime(strict=False)]:
assert (obj.hour, obj.minute, obj.microsecond) == (0, 30, 123456)
assert obj.utcoffset() == hour_delta
def test_type_dts():
actual = temporenc.packb(
type='DTS',
year=1983, month=1, day=15,
hour=18, minute=25, second=12, millisecond=123)
dts_ms = from_hex('47 bf 07 49 93 07 b0')
assert actual == dts_ms
v = temporenc.unpackb(dts_ms)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123000
assert v.nanosecond == 123000000
actual = temporenc.packb(
type='DTS',
year=1983, month=1, day=15,
hour=18, minute=25, second=12, microsecond=123456)
dts_us = from_hex('57 bf 07 49 93 07 89 00')
assert actual == dts_us
v = temporenc.unpackb(dts_us)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123456
assert v.nanosecond == 123456000
actual = temporenc.packb(
type='DTS',
year=1983, month=1, day=15,
hour=18, minute=25, second=12, nanosecond=123456789)
dts_ns = from_hex('67 bf 07 49 93 07 5b cd 15')
assert actual == dts_ns
v = temporenc.unpackb(dts_ns)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123456
assert v.nanosecond == 123456789
actual = temporenc.packb(
type='DTS',
year=1983, month=1, day=15,
hour=18, minute=25, second=12)
dts_none = from_hex('77 bf 07 49 93 00')
assert actual == dts_none
v = temporenc.unpackb(dts_none)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond is None
assert v.microsecond is None
assert v.nanosecond is None
def test_native_unpacking_incomplete():
moment = temporenc.unpackb(temporenc.packb(type='DT', year=1983, hour=12))
with pytest.raises(ValueError):
moment.date()
with pytest.raises(ValueError):
moment.time()
with pytest.raises(ValueError):
moment.datetime()
moment = temporenc.unpackb(temporenc.packb(datetime.datetime.now().date()))
with pytest.raises(ValueError):
moment.datetime()
def test_native_unpacking_incomplete():
moment = temporenc.unpackb(temporenc.packb(type='DT', year=1983, hour=12))
with pytest.raises(ValueError):
moment.date()
with pytest.raises(ValueError):
moment.time()
with pytest.raises(ValueError):
moment.datetime()
moment = temporenc.unpackb(temporenc.packb(datetime.datetime.now().date()))
with pytest.raises(ValueError):
moment.datetime()
def test_incorrect_sizes():
# Too long
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(year=1983) + b'foo')
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(millisecond=0) + b'foo')
# Too short
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(year=1983)[:-1])
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(millisecond=0)[:-1])
def test_incorrect_sizes():
# Too long
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(year=1983) + b'foo')
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(millisecond=0) + b'foo')
# Too short
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(year=1983)[:-1])
with pytest.raises(ValueError):
temporenc.unpackb(temporenc.packb(millisecond=0)[:-1])
def test_type_dtsz():
actual = temporenc.packb(
type='DTSZ',
year=1983, month=1, day=15,
hour=18, minute=25, second=12, millisecond=123,
tz_offset=60)
dtsz_ms = from_hex('e3 df 83 a4 c9 83 dc 40')
assert actual == dtsz_ms
v = temporenc.unpackb(dtsz_ms)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123000
assert v.nanosecond == 123000000
assert v.tz_offset == 60
actual = temporenc.packb(
type='DTSZ',
year=1983, month=1, day=15,
hour=18, minute=25, second=12, microsecond=123456,
tz_offset=60)
dtsz_us = from_hex('eb df 83 a4 c9 83 c4 81 10')
assert actual == dtsz_us
assert temporenc.unpackb(dtsz_us).microsecond == 123456
assert v.tz_offset == 60
actual = temporenc.packb(
type='DTSZ',
year=1983, month=1, day=15,
hour=18, minute=25, second=12, nanosecond=123456789,
tz_offset=60)
dtsz_ns = from_hex('f3 df 83 a4 c9 83 ad e6 8a c4')
assert actual == dtsz_ns
assert temporenc.unpackb(dtsz_ns).nanosecond == 123456789
assert v.tz_offset == 60
actual = temporenc.packb(
type='DTSZ',
year=1983, month=1, day=15,
hour=18, minute=25, second=12,
tz_offset=60)
dtsz_none = from_hex('fb df 83 a4 c9 91 00')
assert actual == dtsz_none
v = temporenc.unpackb(dtsz_none)
assert v.millisecond is None
assert v.millisecond is None
assert v.millisecond is None
assert v.tz_offset == 60
def test_native_time_zone():
# Python < 3.2 doesn't have concrete tzinfo implementations. This
# test uses the internal helper class instead to avoid depending on
# newer Python versions (or on pytz).
from temporenc.temporenc import FixedOffset
dutch_winter = FixedOffset(60) # UTC +01:00
zero_delta = datetime.timedelta(0)
hour_delta = datetime.timedelta(minutes=60)
expected_name = "UTC+01:00"
assert dutch_winter.tzname(None) == expected_name
assert expected_name in str(dutch_winter)
assert expected_name in repr(dutch_winter)
assert dutch_winter.dst(None) == zero_delta
# DTZ
actual = temporenc.packb(
datetime.datetime(1983, 1, 15, 18, 25, 12, 0, tzinfo=dutch_winter),
type='DTZ')
expected = from_hex('cf 7e 0e 93 26 44')
assert actual == expected
moment = temporenc.unpackb(expected)
assert moment.hour == 18
assert moment.tz_offset == 60
dt = moment.datetime()
assert dt.hour == 18
assert dt.utcoffset() == hour_delta
# DTSZ (microsecond, since native types have that precision)
actual = temporenc.packb(
datetime.datetime(
1983, 1, 15, 18, 25, 12, 123456,
tzinfo=dutch_winter),
type='DTSZ')
dtsz_us = from_hex('eb df 83 a4 c9 83 c4 81 10')
assert actual == dtsz_us
moment = temporenc.unpackb(expected)
assert moment.datetime().hour == 18
# Time only with time zone
moment = temporenc.unpackb(temporenc.packb(
datetime.time(0, 30, 0, 123456, tzinfo=dutch_winter),
type='DTSZ'))
assert moment.tz_offset == 60
for obj in [moment.time(), moment.datetime(strict=False)]:
assert (obj.hour, obj.minute, obj.microsecond) == (0, 30, 123456)
assert obj.utcoffset() == hour_delta
def encode(self, serializer: AbstractSerializer) -> RawHeaderBody:
header = [
temporenc.packb(self.created_at),
]
serialized_header: bytes = mpackb(header)
serialized_body: bytes = serializer(self.body)
return RawHeaderBody(serialized_header, serialized_body, None)
def test_hash():
now = datetime.datetime.now()
later = now.replace(microsecond=0) + datetime.timedelta(hours=1)
v1 = temporenc.unpackb(temporenc.packb(now))
v2 = temporenc.unpackb(temporenc.packb(now))
v3 = temporenc.unpackb(temporenc.packb(later))
assert hash(v1) == hash(v2)
assert hash(v1) != hash(v3)
d = {}
d[v1] = 1
d[v2] = 2
d[v3] = 3
assert len(d) == 2
assert d[v1] == 2
def test_native_packing_with_overrides():
actual = temporenc.packb(datetime.datetime(1984, 1, 16, 18, 26, 12,
123456),
year=1983,
day=15,
minute=25)
expected = from_hex('57 bf 07 49 93 07 89 00')
assert actual == expected
def test_hash():
now = datetime.datetime.now()
later = now.replace(microsecond=0) + datetime.timedelta(hours=1)
v1 = temporenc.unpackb(temporenc.packb(now))
v2 = temporenc.unpackb(temporenc.packb(now))
v3 = temporenc.unpackb(temporenc.packb(later))
assert hash(v1) == hash(v2)
assert hash(v1) != hash(v3)
d = {}
d[v1] = 1
d[v2] = 2
d[v3] = 3
assert len(d) == 2
assert d[v1] == 2
def test_native_packing():
with pytest.raises(ValueError):
temporenc.packb(object())
# datetime.date => D
actual = temporenc.packb(datetime.date(1983, 1, 15))
expected = from_hex('8f 7e 0e')
assert actual == expected
# datetime.datetime => DTS, unless told otherwise
actual = temporenc.packb(datetime.datetime(
1983, 1, 15, 18, 25, 12, 123456))
expected = from_hex('57 bf 07 49 93 07 89 00')
assert actual == expected
actual = temporenc.packb(
datetime.datetime(1983, 1, 15, 18, 25, 12),
type='DT')
expected = from_hex('1e fc 1d 26 4c')
assert actual == expected
# datetime.time => DTS, unless told otherwise
assert len(temporenc.packb(datetime.datetime.now().time())) == 8
actual = temporenc.packb(
datetime.time(18, 25, 12),
type='T')
expected = from_hex('a1 26 4c')
assert actual == expected
def test_native_packing():
with pytest.raises(ValueError):
temporenc.packb(object())
# datetime.date => D
actual = temporenc.packb(datetime.date(1983, 1, 15))
expected = from_hex('8f 7e 0e')
assert actual == expected
# datetime.datetime => DTS, unless told otherwise
actual = temporenc.packb(datetime.datetime(1983, 1, 15, 18, 25, 12,
123456))
expected = from_hex('57 bf 07 49 93 07 89 00')
assert actual == expected
actual = temporenc.packb(datetime.datetime(1983, 1, 15, 18, 25, 12),
type='DT')
expected = from_hex('1e fc 1d 26 4c')
assert actual == expected
# datetime.time => DTS, unless told otherwise
assert len(temporenc.packb(datetime.datetime.now().time())) == 8
actual = temporenc.packb(datetime.time(18, 25, 12), type='T')
expected = from_hex('a1 26 4c')
assert actual == expected
def test_type_t():
actual = temporenc.packb(type='T', hour=18, minute=25, second=12)
expected = from_hex('a1 26 4c')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (None, None, None)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
def test_type_d():
actual = temporenc.packb(type='D', year=1983, month=1, day=15)
expected = from_hex('8f 7e 0e')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (None, None, None)
def test_type_t():
actual = temporenc.packb(type='T', hour=18, minute=25, second=12)
expected = from_hex('a1 26 4c')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (None, None, None)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
def test_type_d():
actual = temporenc.packb(type='D', year=1983, month=1, day=15)
expected = from_hex('8f 7e 0e')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (None, None, None)
def test_native_unpacking_leap_second():
value = temporenc.unpackb(temporenc.packb(
year=2013, month=6, day=30,
hour=23, minute=59, second=60))
with pytest.raises(ValueError):
value.datetime() # second out of range
dt = value.datetime(strict=False)
assert dt == datetime.datetime(2013, 6, 30, 23, 59, 59)
def test_type_dt():
actual = temporenc.packb(
type='DT',
year=1983, month=1, day=15,
hour=18, minute=25, second=12)
expected = from_hex('1e fc 1d 26 4c')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
def test_type_dtz():
actual = temporenc.packb(
type='DTZ',
year=1983, month=1, day=15,
hour=18, minute=25, second=12,
tz_offset=60)
expected = from_hex('cf 7e 0e 93 26 44')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.tz_offset == 60
def test_native_unpacking_leap_second():
value = temporenc.unpackb(
temporenc.packb(year=2013,
month=6,
day=30,
hour=23,
minute=59,
second=60))
with pytest.raises(ValueError):
value.datetime() # second out of range
dt = value.datetime(strict=False)
assert dt == datetime.datetime(2013, 6, 30, 23, 59, 59)
def test_native_unpacking():
value = temporenc.unpackb(temporenc.packb(year=1983, month=1, day=15))
assert value.date() == datetime.date(1983, 1, 15)
value = temporenc.unpackb(
temporenc.packb(year=1983,
month=1,
day=15,
hour=1,
minute=2,
second=3,
microsecond=456))
assert value.datetime() == datetime.datetime(1983, 1, 15, 1, 2, 3, 456)
value = temporenc.unpackb(
temporenc.packb(
year=1983,
month=1,
day=15, # will be ignored
hour=1,
minute=2,
second=3,
microsecond=456))
assert value.time() == datetime.time(1, 2, 3, 456)
value = temporenc.unpackb(temporenc.packb(year=1234))
with pytest.raises(ValueError):
value.date()
assert value.date(strict=False).year == 1234
assert value.datetime(strict=False).year == 1234
value = temporenc.unpackb(temporenc.packb(hour=14))
with pytest.raises(ValueError):
value.time()
assert value.time(strict=False).hour == 14
assert value.datetime(strict=False).hour == 14
def test_type_dt():
actual = temporenc.packb(type='DT',
year=1983,
month=1,
day=15,
hour=18,
minute=25,
second=12)
expected = from_hex('1e fc 1d 26 4c')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
def obj_encoder(obj):
if isinstance(obj, typing.Mapping):
return dict(obj)
if isinstance(obj, pd.DataFrame):
return dict(__type__='pandas.DataFrame',
__value__=base64.b64encode(obj.to_msgpack()))
if isinstance(obj, datetime.datetime):
return dict(__type__='temporenc.datetime',
__value__=base64.b64encode(temporenc.packb(obj)))
if isinstance(obj, tuple):
return list(obj)
return obj
def test_type_dtz():
actual = temporenc.packb(type='DTZ',
year=1983,
month=1,
day=15,
hour=18,
minute=25,
second=12,
tz_offset=60)
expected = from_hex('cf 7e 0e 93 26 44')
assert actual == expected
v = temporenc.unpackb(expected)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.tz_offset == 60
def test_string_conversion():
# Date only
value = temporenc.unpackb(temporenc.packb(year=1983, month=1, day=15))
assert str(value) == "1983-01-15"
value = temporenc.unpackb(temporenc.packb(year=1983, day=15))
assert str(value) == "1983-??-15"
# Time only
value = temporenc.unpackb(temporenc.packb(hour=1, minute=2, second=3))
assert str(value) == "01:02:03"
value = temporenc.unpackb(temporenc.packb(
hour=1, second=3, microsecond=12340))
assert str(value) == "01:??:03.01234"
# Date and time
value = temporenc.unpackb(temporenc.packb(
year=1983, month=1, day=15,
hour=18, minute=25))
assert str(value) == "1983-01-15 18:25:??"
# If sub-second is set but equal to 0, the string should show it
# properly anyway.
value = temporenc.unpackb(temporenc.packb(
hour=12, minute=34, second=56, microsecond=0))
assert str(value) == "12:34:56.0"
# Time zone info should be included
moment = temporenc.unpackb(from_hex('cf 7e 0e 93 26 40'))
assert str(moment) == '1983-01-15 18:25:12Z'
moment = temporenc.unpackb(from_hex('cf 7e 0e 93 26 44'))
assert str(moment) == '1983-01-15 18:25:12+01:00'
# Very contrived example...
value = temporenc.unpackb(temporenc.packb(microsecond=1250))
assert str(value) == "??:??:??.00125"
# And a basic one for repr()
value = temporenc.unpackb(temporenc.packb(hour=12, minute=34, second=56))
assert '12:34:56' in repr(value)
def test_string_conversion():
# Date only
value = temporenc.unpackb(temporenc.packb(year=1983, month=1, day=15))
assert str(value) == "1983-01-15"
value = temporenc.unpackb(temporenc.packb(year=1983, day=15))
assert str(value) == "1983-??-15"
# Time only
value = temporenc.unpackb(temporenc.packb(hour=1, minute=2, second=3))
assert str(value) == "01:02:03"
value = temporenc.unpackb(
temporenc.packb(hour=1, second=3, microsecond=12340))
assert str(value) == "01:??:03.01234"
# Date and time
value = temporenc.unpackb(
temporenc.packb(year=1983, month=1, day=15, hour=18, minute=25))
assert str(value) == "1983-01-15 18:25:??"
# If sub-second is set but equal to 0, the string should show it
# properly anyway.
value = temporenc.unpackb(
temporenc.packb(hour=12, minute=34, second=56, microsecond=0))
assert str(value) == "12:34:56.0"
# Time zone info should be included
moment = temporenc.unpackb(from_hex('cf 7e 0e 8b 26 40'))
assert str(moment) == '1983-01-15 17:25:12 (UTC)'
moment = temporenc.unpackb(from_hex('cf 7e 0e 8b 26 44'))
assert str(moment) == '1983-01-15 17:25:12 (UTC; original offset +01:00)'
# Very contrived example...
value = temporenc.unpackb(temporenc.packb(microsecond=1250))
assert str(value) == "??:??:??.00125"
# And a basic one for repr()
value = temporenc.unpackb(temporenc.packb(hour=12, minute=34, second=56))
assert '12:34:56' in repr(value)
def test_wrong_type():
with pytest.raises(ValueError):
temporenc.packb(type="foo", year=1983)
def test_type_dtsz():
# Note: hour is adjusted for UTC
actual = temporenc.packb(type='DTSZ',
year=1983,
month=1,
day=15,
hour=17,
minute=25,
second=12,
millisecond=123,
tz_offset=60)
dtsz_ms = from_hex('e3 df 83 a2 c9 83 dc 40')
assert actual == dtsz_ms
v = temporenc.unpackb(dtsz_ms)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (17, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123000
assert v.nanosecond == 123000000
assert v.tz_offset == 60
actual = temporenc.packb(type='DTSZ',
year=1983,
month=1,
day=15,
hour=17,
minute=25,
second=12,
microsecond=123456,
tz_offset=60)
dtsz_us = from_hex('eb df 83 a2 c9 83 c4 81 10')
assert actual == dtsz_us
assert temporenc.unpackb(dtsz_us).microsecond == 123456
assert v.tz_offset == 60
actual = temporenc.packb(type='DTSZ',
year=1983,
month=1,
day=15,
hour=17,
minute=25,
second=12,
nanosecond=123456789,
tz_offset=60)
dtsz_ns = from_hex('f3 df 83 a2 c9 83 ad e6 8a c4')
assert actual == dtsz_ns
assert temporenc.unpackb(dtsz_ns).nanosecond == 123456789
assert v.tz_offset == 60
actual = temporenc.packb(type='DTSZ',
year=1983,
month=1,
day=15,
hour=17,
minute=25,
second=12,
tz_offset=60)
dtsz_none = from_hex('fb df 83 a2 c9 91 00')
assert actual == dtsz_none
v = temporenc.unpackb(dtsz_none)
assert v.millisecond is None
assert v.millisecond is None
assert v.millisecond is None
assert v.tz_offset == 60
def test_out_of_range_values():
with pytest.raises(ValueError):
temporenc.packb(year=123456)
with pytest.raises(ValueError):
temporenc.packb(month=-12)
with pytest.raises(ValueError):
temporenc.packb(day=1234)
with pytest.raises(ValueError):
temporenc.packb(hour=1234)
with pytest.raises(ValueError):
temporenc.packb(minute=1234)
with pytest.raises(ValueError):
temporenc.packb(second=1234)
with pytest.raises(ValueError):
temporenc.packb(millisecond=1000)
with pytest.raises(ValueError):
temporenc.packb(microsecond=1000000)
with pytest.raises(ValueError):
temporenc.packb(nanosecond=10000000000)
with pytest.raises(ValueError):
temporenc.packb(tz_offset=1050)
with pytest.raises(ValueError):
temporenc.packb(tz_offset=13) # not a full quarter
def test_wrong_type():
with pytest.raises(ValueError):
temporenc.packb(type="foo", year=1983)
def test_type_detection():
# Empty value, so should result in the smallest type
assert len(temporenc.packb()) == 3
# Type D
assert len(temporenc.packb(year=1983)) == 3
assert temporenc.unpackb(temporenc.packb(year=1983)).year == 1983
# Type T
assert len(temporenc.packb(hour=18)) == 3
assert temporenc.unpackb(temporenc.packb(hour=18)).hour == 18
# Type DT
assert len(temporenc.packb(year=1983, hour=18)) == 5
# Type DTS
assert len(temporenc.packb(millisecond=0)) == 7
assert len(temporenc.packb(microsecond=0)) == 8
assert len(temporenc.packb(nanosecond=0)) == 9
# Type DTZ
assert len(temporenc.packb(year=1983, hour=18, tz_offset=120)) == 6
# Type DTSZ
assert len(temporenc.packb(millisecond=0, tz_offset=120)) == 8
def test_native_packing_with_overrides():
actual = temporenc.packb(
datetime.datetime(1984, 1, 16, 18, 26, 12, 123456),
year=1983, day=15, minute=25)
expected = from_hex('57 bf 07 49 93 07 89 00')
assert actual == expected
def test_type_detection():
# Empty value, so should result in the smallest type
assert len(temporenc.packb()) == 3
# Type D
assert len(temporenc.packb(year=1983)) == 3
assert temporenc.unpackb(temporenc.packb(year=1983)).year == 1983
# Type T
assert len(temporenc.packb(hour=18)) == 3
assert temporenc.unpackb(temporenc.packb(hour=18)).hour == 18
# Type DT
assert len(temporenc.packb(year=1983, hour=18)) == 5
# Type DTS
assert len(temporenc.packb(millisecond=0)) == 7
assert len(temporenc.packb(microsecond=0)) == 8
assert len(temporenc.packb(nanosecond=0)) == 9
# Type DTZ
assert len(temporenc.packb(year=1983, hour=18, tz_offset=120)) == 6
# Type DTSZ
assert len(temporenc.packb(millisecond=0, tz_offset=120)) == 8
def test_type_empty_values():
v = temporenc.unpackb(temporenc.packb(type='DTS'))
assert (v.year, v.month, v.day) == (None, None, None)
assert (v.hour, v.minute, v.second) == (None, None, None)
assert (v.millisecond, v.microsecond, v.nanosecond) == (None, None, None)
assert v.tz_offset is None
def test_out_of_range_values():
with pytest.raises(ValueError):
temporenc.packb(year=123456)
with pytest.raises(ValueError):
temporenc.packb(month=-12)
with pytest.raises(ValueError):
temporenc.packb(day=1234)
with pytest.raises(ValueError):
temporenc.packb(hour=1234)
with pytest.raises(ValueError):
temporenc.packb(minute=1234)
with pytest.raises(ValueError):
temporenc.packb(second=1234)
with pytest.raises(ValueError):
temporenc.packb(millisecond=1000)
with pytest.raises(ValueError):
temporenc.packb(microsecond=1000000)
with pytest.raises(ValueError):
temporenc.packb(nanosecond=10000000000)
with pytest.raises(ValueError):
temporenc.packb(tz_offset=1050)
with pytest.raises(ValueError):
temporenc.packb(tz_offset=13) # not a full quarter
def test_type_empty_values():
v = temporenc.unpackb(temporenc.packb(type='DTS'))
assert (v.year, v.month, v.day) == (None, None, None)
assert (v.hour, v.minute, v.second) == (None, None, None)
assert (v.millisecond, v.microsecond, v.nanosecond) == (None, None, None)
assert v.tz_offset is None
def serialize(self, obj: datetime, key_size: int) -> bytes:
if obj.tzinfo is None:
raise ValueError("DatetimeUTCSerializer needs a timezone aware"
" datetime")
return temporenc.packb(obj, type="DTS")
def test_type_dts():
actual = temporenc.packb(type='DTS',
year=1983,
month=1,
day=15,
hour=18,
minute=25,
second=12,
millisecond=123)
dts_ms = from_hex('47 bf 07 49 93 07 b0')
assert actual == dts_ms
v = temporenc.unpackb(dts_ms)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123000
assert v.nanosecond == 123000000
actual = temporenc.packb(type='DTS',
year=1983,
month=1,
day=15,
hour=18,
minute=25,
second=12,
microsecond=123456)
dts_us = from_hex('57 bf 07 49 93 07 89 00')
assert actual == dts_us
v = temporenc.unpackb(dts_us)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123456
assert v.nanosecond == 123456000
actual = temporenc.packb(type='DTS',
year=1983,
month=1,
day=15,
hour=18,
minute=25,
second=12,
nanosecond=123456789)
dts_ns = from_hex('67 bf 07 49 93 07 5b cd 15')
assert actual == dts_ns
v = temporenc.unpackb(dts_ns)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond == 123
assert v.microsecond == 123456
assert v.nanosecond == 123456789
actual = temporenc.packb(type='DTS',
year=1983,
month=1,
day=15,
hour=18,
minute=25,
second=12)
dts_none = from_hex('77 bf 07 49 93 00')
assert actual == dts_none
v = temporenc.unpackb(dts_none)
assert (v.year, v.month, v.day) == (1983, 1, 15)
assert (v.hour, v.minute, v.second) == (18, 25, 12)
assert v.millisecond is None
assert v.microsecond is None
assert v.nanosecond is None
