mirror of
https://github.com/kennethreitz/maya.git
synced 2026-06-05 23:00:18 +00:00
Jan Vlcinsky
e9592a3146
repeated code replaced by parameters provided by pytest. source string and rexpected intentionally indented to match by column to make checking year, month and date simpler. Note, that the test_issue_104 was (and still is) failing.
315 lines
7.8 KiB
Python
315 lines
7.8 KiB
Python
import copy
|
|
import time
|
|
from datetime import timedelta, datetime as Datetime
|
|
|
|
import pytz
|
|
import pytest
|
|
|
|
import maya
|
|
from maya.core import _seconds_or_timedelta # import private function
|
|
|
|
|
|
@pytest.mark.parametrize("string,expected", [
|
|
('February 21, 1994',
|
|
'Mon, 21 Feb 1994 00:00:00 GMT'),
|
|
])
|
|
def test_rfc2822(string, expected):
|
|
r = maya.parse(string).rfc2822()
|
|
d = maya.MayaDT.from_rfc2822(r)
|
|
assert r == expected
|
|
assert r == d.rfc2822()
|
|
|
|
|
|
@pytest.mark.parametrize("string,expected", [
|
|
('February 21, 1994',
|
|
'1994-02-21T00:00:00Z'),
|
|
])
|
|
def test_iso8601(string, expected):
|
|
r = maya.parse(string).iso8601()
|
|
d = maya.MayaDT.from_iso8601(r)
|
|
assert r == expected
|
|
assert r == d.iso8601()
|
|
|
|
|
|
@pytest.mark.parametrize("string,expected", [
|
|
('20161001T1430.4+05:30',
|
|
'2016-10-01T09:00:00.400000Z'),
|
|
|
|
('2016T14',
|
|
'2016-01-01T14:00:00Z'),
|
|
|
|
('2016-10T14',
|
|
'2016-10-01T14:00:00Z'),
|
|
|
|
('2012W05',
|
|
'2012-01-30T00:00:00Z'),
|
|
|
|
('2012W055',
|
|
'2012-02-03T00:00:00Z'),
|
|
|
|
('2012007',
|
|
'2012-01-07T00:00:00Z'),
|
|
|
|
('2016-W07T09',
|
|
'2016-02-15T09:00:00Z'),
|
|
])
|
|
def test_parse_iso8601(string, expected):
|
|
d = maya.MayaDT.from_iso8601(string)
|
|
assert expected == d.iso8601()
|
|
|
|
|
|
def test_struct():
|
|
now = round(time.time())
|
|
ts = time.gmtime(now)
|
|
m = maya.MayaDT.from_struct(ts)
|
|
dt = Datetime.fromtimestamp(now, pytz.UTC)
|
|
assert m._epoch is not None
|
|
assert m.datetime() == dt
|
|
ts = time.localtime(now)
|
|
m = maya.MayaDT.from_struct(ts)
|
|
dt = Datetime.fromtimestamp(
|
|
time.mktime(ts) - maya.core.utc_offset(), pytz.UTC
|
|
)
|
|
assert m._epoch is not None
|
|
assert m.datetime() == dt
|
|
|
|
|
|
def test_issue_104():
|
|
e = 1507756331
|
|
t = Datetime.utcfromtimestamp(e)
|
|
t = maya.MayaDT.from_datetime(t)
|
|
assert str(t) == 'Wed, 11 Oct 2017 21:12:11 GMT'
|
|
t = time.gmtime(e)
|
|
t = maya.MayaDT.from_struct(t)
|
|
assert str(t) == 'Wed, 11 Oct 2017 21:12:11 GMT'
|
|
|
|
|
|
def test_human_when():
|
|
r1 = maya.when('yesterday')
|
|
r2 = maya.when('today')
|
|
assert (r2.day - r1.day) in (1, -30, -29, -28, -27)
|
|
|
|
|
|
def test_machine_parse():
|
|
r1 = maya.parse('August 14, 2015')
|
|
assert r1.day == 14
|
|
r2 = maya.parse('August 15, 2015')
|
|
assert r2.day == 15
|
|
|
|
|
|
def test_dt_tz_translation():
|
|
d1 = maya.now().datetime()
|
|
d2 = maya.now().datetime(to_timezone='EST')
|
|
assert (d1.hour - d2.hour) % 24 == 5
|
|
|
|
|
|
def test_dt_tz_naive():
|
|
d1 = maya.now().datetime(naive=True)
|
|
assert d1.tzinfo is None
|
|
d2 = maya.now().datetime(to_timezone='EST', naive=True)
|
|
assert d2.tzinfo is None
|
|
assert (d1.hour - d2.hour) % 24 == 5
|
|
|
|
|
|
def test_random_date():
|
|
# Test properties for maya.when()
|
|
d1 = maya.when('11-17-11 08:09:10')
|
|
assert d1.year == 2011
|
|
assert d1.month == 11
|
|
assert d1.day == 17
|
|
assert d1.week == 46
|
|
assert d1.weekday == 4
|
|
assert d1.hour == 8
|
|
assert d1.minute == 9
|
|
assert d1.second == 10
|
|
assert d1.microsecond == 0
|
|
# Test properties for maya.parse()
|
|
d2 = maya.parse('February 29, 1992 13:12:34')
|
|
assert d2.year == 1992
|
|
assert d2.month == 2
|
|
assert d2.day == 29
|
|
assert d2.week == 9
|
|
assert d2.weekday == 6
|
|
assert d2.hour == 13
|
|
assert d2.minute == 12
|
|
assert d2.second == 34
|
|
assert d2.microsecond == 0
|
|
|
|
|
|
def test_print_date(capsys):
|
|
d = maya.when('11-17-11')
|
|
print(d)
|
|
out, err = capsys.readouterr()
|
|
assert out == 'Thu, 17 Nov 2011 00:00:00 GMT\n'
|
|
assert repr(d) == '<MayaDT epoch=1321488000.0>'
|
|
|
|
|
|
def test_invalid_date():
|
|
with pytest.raises(ValueError):
|
|
maya.when('another day')
|
|
|
|
|
|
def test_slang_date():
|
|
d = maya.when('tomorrow')
|
|
assert d.slang_date() == 'tomorrow'
|
|
|
|
|
|
def test_slang_time():
|
|
d = maya.when('1 hour ago')
|
|
assert d.slang_time() == 'an hour ago'
|
|
|
|
|
|
@pytest.mark.parametrize("string,kwds,expected", [
|
|
('February 21, 1994', {},
|
|
'1994-02-21 00:00:00+00:00'),
|
|
|
|
('01/05/2016', {},
|
|
'2016-01-05 00:00:00+00:00'),
|
|
|
|
('01/05/2016', dict(day_first=True),
|
|
'2016-05-01 00:00:00+00:00'),
|
|
|
|
('2016/05/01', dict(year_first=True, day_first=False),
|
|
'2016-05-01 00:00:00+00:00'),
|
|
|
|
('2016/01/05', dict(year_first=True, day_first=True),
|
|
'2016-05-01 00:00:00+00:00'),
|
|
|
|
('01/05/2016', dict(timezone='UTC'),
|
|
'2016-01-05 00:00:00+00:00'),
|
|
|
|
('01/05/2016', dict(timezone='US/Central'),
|
|
'2016-01-05 06:00:00+00:00'),
|
|
])
|
|
def test_parse(string, kwds, expected):
|
|
d = maya.parse(string, **kwds)
|
|
assert format(d) == expected
|
|
|
|
|
|
def test_when_past():
|
|
next_month = str(maya.now().add(months=1).month)
|
|
this_year = maya.now().year
|
|
last_year = this_year - 1
|
|
future_date = maya.when(next_month)
|
|
past_date = maya.when(next_month, prefer_past=True)
|
|
assert future_date.year == this_year
|
|
if next_month == '1':
|
|
assert past_date.year == this_year
|
|
else:
|
|
assert past_date.year == last_year
|
|
|
|
|
|
def test_datetime_to_timezone():
|
|
dt = maya.when('2016-01-01').datetime(to_timezone='US/Eastern')
|
|
assert dt.tzinfo.zone == 'US/Eastern'
|
|
|
|
|
|
def test_rfc3339():
|
|
mdt = maya.when('2016-01-01')
|
|
out = mdt.rfc3339()
|
|
mdt2 = maya.MayaDT.from_rfc3339(out)
|
|
assert mdt.epoch == mdt2.epoch
|
|
|
|
|
|
def test_comparison_operations():
|
|
now = maya.now()
|
|
now_copy = copy.deepcopy(now)
|
|
tomorrow = maya.when('tomorrow')
|
|
assert (now == now_copy) is True
|
|
assert (now == tomorrow) is False
|
|
assert (now != now_copy) is False
|
|
assert (now != tomorrow) is True
|
|
assert (now < now_copy) is False
|
|
assert (now < tomorrow) is True
|
|
assert (now <= now_copy) is True
|
|
assert (now <= tomorrow) is True
|
|
assert (now > now_copy) is False
|
|
assert (now > tomorrow) is False
|
|
assert (now >= now_copy) is True
|
|
assert (now >= tomorrow) is False
|
|
# Check Exceptions
|
|
with pytest.raises(TypeError):
|
|
now == 1
|
|
with pytest.raises(TypeError):
|
|
now != 1
|
|
with pytest.raises(TypeError):
|
|
now < 1
|
|
with pytest.raises(TypeError):
|
|
now <= 1
|
|
with pytest.raises(TypeError):
|
|
now > 1
|
|
with pytest.raises(TypeError):
|
|
now >= 1
|
|
|
|
|
|
def test_seconds_or_timedelta():
|
|
# test for value in seconds
|
|
assert _seconds_or_timedelta(1234) == timedelta(0, 1234)
|
|
# test for value as `datetime.timedelta`
|
|
assert _seconds_or_timedelta(timedelta(0, 1234)) == timedelta(0, 1234)
|
|
# test for invalid value
|
|
with pytest.raises(TypeError):
|
|
_seconds_or_timedelta('invalid interval')
|
|
|
|
|
|
def test_intervals():
|
|
now = maya.now()
|
|
tomorrow = now.add(days=1)
|
|
assert len(list(maya.intervals(now, tomorrow, 60 * 60))) == 24
|
|
|
|
|
|
def test_dunder_add():
|
|
now = maya.now()
|
|
assert now + 1 == now.add(seconds=1)
|
|
assert now + timedelta(seconds=1) == now.add(seconds=1)
|
|
|
|
|
|
def test_dunder_radd():
|
|
now = maya.now()
|
|
assert now.add(seconds=1) == now + 1
|
|
assert now.add(seconds=1) == now + timedelta(seconds=1)
|
|
|
|
|
|
def test_dunder_sub():
|
|
now = maya.now()
|
|
assert now - 1 == now.subtract(seconds=1)
|
|
assert now - timedelta(seconds=1) == now.subtract(seconds=1)
|
|
|
|
|
|
def test_mayaDT_sub():
|
|
now = maya.now()
|
|
then = now.add(days=1)
|
|
assert then - now == timedelta(24 * 60 * 60)
|
|
assert now - then == timedelta(-24 * 60 * 60)
|
|
|
|
|
|
def test_core_local_timezone(monkeypatch):
|
|
|
|
@property
|
|
def mock_local_tz(self):
|
|
|
|
class StaticTzInfo(object):
|
|
zone = 'local'
|
|
|
|
def __repr__(self):
|
|
return "<StaticTzInfo 'local'>"
|
|
|
|
return StaticTzInfo()
|
|
|
|
monkeypatch.setattr(maya.MayaDT, '_local_tz', mock_local_tz)
|
|
mdt = maya.MayaDT(0)
|
|
assert mdt.local_timezone == 'UTC'
|
|
|
|
|
|
@pytest.mark.parametrize("when_str,snap_str,expected_when", [
|
|
('Mon, 21 Feb 1994 21:21:42 GMT', '@d',
|
|
'Mon, 21 Feb 1994 00:00:00 GMT'),
|
|
])
|
|
def test_snaptime(when_str, snap_str, expected_when):
|
|
# given
|
|
dt = maya.when(when_str)
|
|
# when
|
|
dt = dt.snap(snap_str)
|
|
# then
|
|
assert dt == maya.when(expected_when)
|