maya/tests/test_maya.py

import copy
import time
import calendar
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()


@pytest.mark.usefixtures("frozen_now")
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(ts), 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


@pytest.mark.usefixtures("frozen_now")
def test_dt_tz_translation():
    d1 = maya.now().datetime()
    d2 = maya.now().datetime(to_timezone='EST')
    assert (d1.hour - d2.hour) % 24 == 5


@pytest.mark.usefixtures("frozen_now")
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_date_locale():
    d = maya.when('tomorrow')
    assert d.slang_date(locale='fr') == 'demain'


def test_slang_time():
    d = maya.when('1 hour ago')
    assert d.slang_time() == '1 hour ago'


def test_slang_time_locale():
    d = maya.when('1 hour ago')
    assert d.slang_time(locale='de') == 'vor 1 Stunde'


@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


@pytest.mark.usefixtures("frozen_now")
def test_when_past():
    two_days_away = maya.now().add(days=2)

    past_date = maya.when(
            two_days_away.slang_date(),
            prefer_dates_from='past')

    assert past_date < maya.now()


@pytest.mark.usefixtures("frozen_now")
def test_when_future():
    two_days_away = maya.now().add(days=2)

    future_date = maya.when(
            two_days_away.slang_date(),
            prefer_dates_from='future')

    assert future_date > maya.now()


@pytest.mark.usefixtures("frozen_now")
def test_when_past_day_name():
    two_days_away = maya.now().add(days=2)

    past_date = maya.when(
            calendar.day_name[two_days_away.weekday],
            prefer_dates_from='past')

    assert past_date < maya.now()


@pytest.mark.usefixtures("frozen_now")
def test_when_future_day_name():
    two_days_away = maya.now().add(days=2)

    future_date = maya.when(
            calendar.day_name[two_days_away.weekday],
            prefer_dates_from='future')

    assert future_date > maya.now()


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_epoch():
    mdt = maya.when('2016-01-01')
    out = mdt.rfc3339()
    mdt2 = maya.MayaDT.from_rfc3339(out)
    assert mdt.epoch == mdt2.epoch


def test_rfc3339_format():
    rfc3339 = maya.MayaDT.rfc3339(maya.when('2016-01-01T12:03:03Z'))
    # it's important that the string has got a "max 1-digit millis" fragment
    # as per https://tools.ietf.org/html/rfc3339#section-5.6
    assert rfc3339 == '2016-01-01T12:03:03.0Z'


@pytest.mark.usefixtures("frozen_now")
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')


@pytest.mark.usefixtures("frozen_now")
def test_intervals():
    now = maya.now()
    tomorrow = now.add(days=1)
    assert len(list(maya.intervals(now, tomorrow, 60 * 60))) == 24


@pytest.mark.usefixtures("frozen_now")
def test_dunder_add():
    now = maya.now()
    assert now + 1 == now.add(seconds=1)
    assert now + timedelta(seconds=1) == now.add(seconds=1)


@pytest.mark.usefixtures("frozen_now")
def test_dunder_radd():
    now = maya.now()
    assert now.add(seconds=1) == now + 1
    assert now.add(seconds=1) == now + timedelta(seconds=1)


@pytest.mark.usefixtures("frozen_now")
def test_dunder_sub():
    now = maya.now()
    assert now - 1 == now.subtract(seconds=1)
    assert now - timedelta(seconds=1) == now.subtract(seconds=1)


@pytest.mark.usefixtures("frozen_now")
def test_mayaDT_sub():
    now = maya.now()
    then = now.add(days=1)
    assert then - now == timedelta(seconds=24 * 60 * 60)
    assert now - then == timedelta(seconds=-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'


def test_getting_datetime_for_local_timezone(monkeypatch):

    @property
    def mock_local_tz(self):
        class StaticTzInfo(object):
            zone = 'Europe/Zurich'

            def __repr__(self):
                return "<StaticTzInfo 'Europe/Zurich'>"

        return StaticTzInfo()

    monkeypatch.setattr(maya.MayaDT, '_local_tz', mock_local_tz)

    d = maya.parse('1994-02-21T12:00:00+05:30')

    dt = pytz.timezone('Europe/Zurich').localize(
            Datetime(1994, 2, 21, 7, 30))

    assert d.local_datetime() == dt


@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)