64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
|
from datetime import (
|
||
|
|
date,
|
||
|
|
datetime,
|
||
|
|
)
|
||
|
|
|
||
|
|
from hypothesis import given
|
||
|
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from pandas._libs.tslibs import ccalendar
|
||
|
|
|
||
|
|
from pandas._testing._hypothesis import DATETIME_IN_PD_TIMESTAMP_RANGE_NO_TZ
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"date_tuple,expected",
|
||
|
|
[
|
||
|
|
((2001, 3, 1), 60),
|
||
|
|
((2004, 3, 1), 61),
|
||
|
|
((1907, 12, 31), 365), # End-of-year, non-leap year.
|
||
|
|
((2004, 12, 31), 366), # End-of-year, leap year.
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_get_day_of_year_numeric(date_tuple, expected):
|
||
|
|
assert ccalendar.get_day_of_year(*date_tuple) == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_day_of_year_dt():
|
||
|
|
dt = datetime.fromordinal(1 + np.random.default_rng(2).integers(365 * 4000))
|
||
|
|
result = ccalendar.get_day_of_year(dt.year, dt.month, dt.day)
|
||
|
|
|
||
|
|
expected = (dt - dt.replace(month=1, day=1)).days + 1
|
||
|
|
assert result == expected
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"input_date_tuple, expected_iso_tuple",
|
||
|
|
[
|
||
|
|
[(2020, 1, 1), (2020, 1, 3)],
|
||
|
|
[(2019, 12, 31), (2020, 1, 2)],
|
||
|
|
[(2019, 12, 30), (2020, 1, 1)],
|
||
|
|
[(2009, 12, 31), (2009, 53, 4)],
|
||
|
|
[(2010, 1, 1), (2009, 53, 5)],
|
||
|
|
[(2010, 1, 3), (2009, 53, 7)],
|
||
|
|
[(2010, 1, 4), (2010, 1, 1)],
|
||
|
|
[(2006, 1, 1), (2005, 52, 7)],
|
||
|
|
[(2005, 12, 31), (2005, 52, 6)],
|
||
|
|
[(2008, 12, 28), (2008, 52, 7)],
|
||
|
|
[(2008, 12, 29), (2009, 1, 1)],
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_dt_correct_iso_8601_year_week_and_day(input_date_tuple, expected_iso_tuple):
|
||
|
|
result = ccalendar.get_iso_calendar(*input_date_tuple)
|
||
|
|
expected_from_date_isocalendar = date(*input_date_tuple).isocalendar()
|
||
|
|
assert result == expected_from_date_isocalendar
|
||
|
|
assert result == expected_iso_tuple
|
||
|
|
|
||
|
|
|
||
|
|
@given(DATETIME_IN_PD_TIMESTAMP_RANGE_NO_TZ)
|
||
|
|
def test_isocalendar(dt):
|
||
|
|
expected = dt.isocalendar()
|
||
|
|
result = ccalendar.get_iso_calendar(dt.year, dt.month, dt.day)
|
||
|
|
assert result == expected
|