diff options
Diffstat (limited to 'datamaps/tests/test_api.py')
-rw-r--r-- | datamaps/tests/test_api.py | 31 |
1 files changed, 0 insertions, 31 deletions
diff --git a/datamaps/tests/test_api.py b/datamaps/tests/test_api.py index ecb30ac..4c73931 100644 --- a/datamaps/tests/test_api.py +++ b/datamaps/tests/test_api.py @@ -32,34 +32,3 @@ def test_get_project_data_using_month(master): assert m2.quarter.end_date == datetime.date(2021, 9, 30) assert m3.quarter.end_date == datetime.date(2021, 9, 30) assert m4.quarter.end_date == datetime.date(2021, 12, 31) - - -def test_quarter_objects_have_months(): - q = Quarter(1, 2021) - q2 = Quarter(4, 2021) - assert isinstance(q.months[0], Month) - 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(): - m1 = Month(1, 2021) - assert m1.name == "January" - m2 = Month(9, 2021) - assert m2.name == "September" - assert m2.start_date == datetime.date(2021, 9, 1) - assert m2.end_date == datetime.date(2021, 9, 30) - # test leap year - m3 = Month(2, 2024) - m4 = Month(2, 2028) - assert m3.end_date == datetime.date(2024, 2, 29) - assert m4.end_date == datetime.date(2028, 2, 29) - |