nearly got tests passing - need to get correct item from list

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():