# -*- coding: utf-8 -*-
from datetime import timedelta

import pytest
import numpy as np

import pandas as pd
import pandas.util.testing as tm
from pandas import Timedelta


def test_construction():
    expected = np.timedelta64(10, 'D').astype('m8[ns]').view('i8')
    assert Timedelta(10, unit='d').value == expected
    assert Timedelta(10.0, unit='d').value == expected
    assert Timedelta('10 days').value == expected
    assert Timedelta(days=10).value == expected
    assert Timedelta(days=10.0).value == expected

    expected += np.timedelta64(10, 's').astype('m8[ns]').view('i8')
    assert Timedelta('10 days 00:00:10').value == expected
    assert Timedelta(days=10, seconds=10).value == expected
    assert Timedelta(days=10, milliseconds=10 * 1000).value == expected
    assert Timedelta(days=10,
                     microseconds=10 * 1000 * 1000).value == expected

    # rounding cases
    assert Timedelta(82739999850000).value == 82739999850000
    assert ('0 days 22:58:59.999850' in str(Timedelta(82739999850000)))
    assert Timedelta(123072001000000).value == 123072001000000
    assert ('1 days 10:11:12.001' in str(Timedelta(123072001000000)))

    # string conversion with/without leading zero
    # GH#9570
    assert Timedelta('0:00:00') == timedelta(hours=0)
    assert Timedelta('00:00:00') == timedelta(hours=0)
    assert Timedelta('-1:00:00') == -timedelta(hours=1)
    assert Timedelta('-01:00:00') == -timedelta(hours=1)

    # more strings & abbrevs
    # GH#8190
    assert Timedelta('1 h') == timedelta(hours=1)
    assert Timedelta('1 hour') == timedelta(hours=1)
    assert Timedelta('1 hr') == timedelta(hours=1)
    assert Timedelta('1 hours') == timedelta(hours=1)
    assert Timedelta('-1 hours') == -timedelta(hours=1)
    assert Timedelta('1 m') == timedelta(minutes=1)
    assert Timedelta('1.5 m') == timedelta(seconds=90)
    assert Timedelta('1 minute') == timedelta(minutes=1)
    assert Timedelta('1 minutes') == timedelta(minutes=1)
    assert Timedelta('1 s') == timedelta(seconds=1)
    assert Timedelta('1 second') == timedelta(seconds=1)
    assert Timedelta('1 seconds') == timedelta(seconds=1)
    assert Timedelta('1 ms') == timedelta(milliseconds=1)
    assert Timedelta('1 milli') == timedelta(milliseconds=1)
    assert Timedelta('1 millisecond') == timedelta(milliseconds=1)
    assert Timedelta('1 us') == timedelta(microseconds=1)
    assert Timedelta('1 micros') == timedelta(microseconds=1)
    assert Timedelta('1 microsecond') == timedelta(microseconds=1)
    assert Timedelta('1.5 microsecond') == Timedelta('00:00:00.000001500')
    assert Timedelta('1 ns') == Timedelta('00:00:00.000000001')
    assert Timedelta('1 nano') == Timedelta('00:00:00.000000001')
    assert Timedelta('1 nanosecond') == Timedelta('00:00:00.000000001')

    # combos
    assert Timedelta('10 days 1 hour') == timedelta(days=10, hours=1)
    assert Timedelta('10 days 1 h') == timedelta(days=10, hours=1)
    assert Timedelta('10 days 1 h 1m 1s') == timedelta(
        days=10, hours=1, minutes=1, seconds=1)
    assert Timedelta('-10 days 1 h 1m 1s') == -timedelta(
        days=10, hours=1, minutes=1, seconds=1)
    assert Timedelta('-10 days 1 h 1m 1s') == -timedelta(
        days=10, hours=1, minutes=1, seconds=1)
    assert Timedelta('-10 days 1 h 1m 1s 3us') == -timedelta(
        days=10, hours=1, minutes=1, seconds=1, microseconds=3)
    assert Timedelta('-10 days 1 h 1.5m 1s 3us') == -timedelta(
        days=10, hours=1, minutes=1, seconds=31, microseconds=3)

    # Currently invalid as it has a - on the hh:mm:dd part
    # (only allowed on the days)
    with pytest.raises(ValueError):
        Timedelta('-10 days -1 h 1.5m 1s 3us')

    # only leading neg signs are allowed
    with pytest.raises(ValueError):
        Timedelta('10 days -1 h 1.5m 1s 3us')

    # no units specified
    with pytest.raises(ValueError):
        Timedelta('3.1415')

    # invalid construction
    tm.assert_raises_regex(ValueError, "cannot construct a Timedelta",
                           lambda: Timedelta())
    tm.assert_raises_regex(ValueError,
                           "unit abbreviation w/o a number",
                           lambda: Timedelta('foo'))
    tm.assert_raises_regex(ValueError,
                           "cannot construct a Timedelta from the "
                           "passed arguments, allowed keywords are ",
                           lambda: Timedelta(day=10))

    # floats
    expected = np.timedelta64(
        10, 's').astype('m8[ns]').view('i8') + np.timedelta64(
            500, 'ms').astype('m8[ns]').view('i8')
    assert Timedelta(10.5, unit='s').value == expected

    # offset
    assert pd.to_timedelta(pd.offsets.Hour(2)) == Timedelta(hours=2)
    assert Timedelta(pd.offsets.Hour(2)) == Timedelta(hours=2)
    assert Timedelta(pd.offsets.Second(2)) == Timedelta(seconds=2)

    # GH#11995: unicode
    expected = Timedelta('1H')
    result = pd.Timedelta(u'1H')
    assert result == expected
    assert (pd.to_timedelta(pd.offsets.Hour(2)) ==
            Timedelta(u'0 days, 02:00:00'))

    with pytest.raises(ValueError):
        Timedelta(u'foo bar')


@pytest.mark.parametrize('item', list({'days': 'D',
                                       'seconds': 's',
                                       'microseconds': 'us',
                                       'milliseconds': 'ms',
                                       'minutes': 'm',
                                       'hours': 'h',
                                       'weeks': 'W'}.items()))
@pytest.mark.parametrize('npdtype', [np.int64, np.int32, np.int16,
                                     np.float64, np.float32, np.float16])
def test_td_construction_with_np_dtypes(npdtype, item):
    # GH#8757: test construction with np dtypes
    pykwarg, npkwarg = item
    expected = np.timedelta64(1, npkwarg).astype('m8[ns]').view('i8')
    assert Timedelta(**{pykwarg: npdtype(1)}).value == expected


@pytest.mark.parametrize('val', [
    '1s', '-1s', '1us', '-1us', '1 day', '-1 day',
    '-23:59:59.999999', '-1 days +23:59:59.999999', '-1ns',
    '1ns', '-23:59:59.999999999'])
def test_td_from_repr_roundtrip(val):
    # round-trip both for string and value
    td = Timedelta(val)
    assert Timedelta(td.value) == td

    # str does not normally display nanos
    if not td.nanoseconds:
        assert Timedelta(str(td)) == td
    assert Timedelta(td._repr_base(format='all')) == td


def test_overflow_on_construction():
    # xref https://github.com/statsmodels/statsmodels/issues/3374
    value = pd.Timedelta('1day').value * 20169940
    with pytest.raises(OverflowError):
        pd.Timedelta(value)

    # xref GH#17637
    with pytest.raises(OverflowError):
        pd.Timedelta(7 * 19999, unit='D')

    with pytest.raises(OverflowError):
        pd.Timedelta(timedelta(days=13 * 19999))


@pytest.mark.parametrize('fmt,exp', [
    ('P6DT0H50M3.010010012S', Timedelta(days=6, minutes=50, seconds=3,
                                        milliseconds=10, microseconds=10,
                                        nanoseconds=12)),
    ('P-6DT0H50M3.010010012S', Timedelta(days=-6, minutes=50, seconds=3,
                                         milliseconds=10, microseconds=10,
                                         nanoseconds=12)),
    ('P4DT12H30M5S', Timedelta(days=4, hours=12, minutes=30, seconds=5)),
    ('P0DT0H0M0.000000123S', Timedelta(nanoseconds=123)),
    ('P0DT0H0M0.00001S', Timedelta(microseconds=10)),
    ('P0DT0H0M0.001S', Timedelta(milliseconds=1)),
    ('P0DT0H1M0S', Timedelta(minutes=1)),
    ('P1DT25H61M61S', Timedelta(days=1, hours=25, minutes=61, seconds=61))
])
def test_iso_constructor(fmt, exp):
    assert Timedelta(fmt) == exp


@pytest.mark.parametrize('fmt', [
    'PPPPPPPPPPPP', 'PDTHMS', 'P0DT999H999M999S',
    'P1DT0H0M0.0000000000000S', 'P1DT0H0M00000000000S',
    'P1DT0H0M0.S'])
def test_iso_constructor_raises(fmt):
    with tm.assert_raises_regex(ValueError, 'Invalid ISO 8601 Duration '
                                'format - {}'.format(fmt)):
        Timedelta(fmt)


def test_td_constructor_on_nanoseconds():
    # GH#9273
    result = Timedelta(nanoseconds=100)
    expected = Timedelta('100ns')
    assert result == expected

    result = Timedelta(days=1, hours=1, minutes=1, weeks=1, seconds=1,
                       milliseconds=1, microseconds=1, nanoseconds=1)
    expected = Timedelta(694861001001001)
    assert result == expected

    result = Timedelta(microseconds=1) + Timedelta(nanoseconds=1)
    expected = Timedelta('1us1ns')
    assert result == expected

    result = Timedelta(microseconds=1) - Timedelta(nanoseconds=1)
    expected = Timedelta('999ns')
    assert result == expected

    result = Timedelta(microseconds=1) + 5 * Timedelta(nanoseconds=-2)
    expected = Timedelta('990ns')
    assert result == expected

    with pytest.raises(TypeError):
        Timedelta(nanoseconds='abc')
