1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
| 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.randint(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
|
|
