diff options
Diffstat (limited to 'datamaps/tests')
| -rw-r--r-- | datamaps/tests/test_api.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/datamaps/tests/test_api.py b/datamaps/tests/test_api.py index 21ab4d1..347ea4a 100644 --- a/datamaps/tests/test_api.py +++ b/datamaps/tests/test_api.py @@ -1,4 +1,5 @@ import datetime +import pytest from ..api import project_data_from_master, project_data_from_master_month from ..core.temporal import Quarter, Month @@ -25,7 +26,17 @@ def test_get_project_data_using_month(master): def test_quarter_objects_have_months(): q = Quarter(1, 2021) + q2 = Quarter(4, 2021) assert q.months[0].start_date == datetime.date(2021, 4, 1) + assert q.months[1].start_date == datetime.date(2021, 5, 1) + assert q.months[2].start_date == datetime.date(2021, 6, 1) + with pytest.raises(IndexError): + q.months[4].start_date == datetime.date(2021, 6, 1) + assert q2.months[0].start_date == datetime.date(2022, 1, 1) + assert q2.months[1].start_date == datetime.date(2022, 2, 1) + assert q2.months[2].start_date == datetime.date(2022, 3, 1) + with pytest.raises(IndexError): + q2.months[4].start_date == datetime.date(2022, 6, 1) def test_month(): |