aboutsummaryrefslogtreecommitdiffstats
path: root/datamaps/tests
diff options
context:
space:
mode:
authorMatthew Lemon <matt@matthewlemon.com>2021-05-19 17:42:17 +0100
committerMatthew Lemon <matt@matthewlemon.com>2021-05-19 17:42:25 +0100
commitf07fd91889749bec2b7a5fce3af4f10b3e2b3f73 (patch)
treefc639089aec8c6a4a24889053f7ef1ad7915af46 /datamaps/tests
parent2ed09e9b628ec32131e981d467591d20acd1455d (diff)
updated repr() for Master associated with a month
Diffstat (limited to '')
-rw-r--r--datamaps/tests/test_api.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/datamaps/tests/test_api.py b/datamaps/tests/test_api.py
index 295f1fb..04a6805 100644
--- a/datamaps/tests/test_api.py
+++ b/datamaps/tests/test_api.py
@@ -19,6 +19,13 @@ def test_get_project_data_using_month(master):
m3 = project_data_from_master_month(master, 9, 2021)
m4 = project_data_from_master_month(master, 10, 2021)
m5 = project_data_from_master_month(master, 2, 2021) # this is q4
+ m6 = project_data_from_master_month(master, 5, 2030) # this is q4
+ assert "Master for Month(July) - 2021" in repr(m)
+ assert "Master for Month(August) - 2021" in repr(m2)
+ assert "Master for Month(September) - 2021" in repr(m3)
+ assert "Master for Month(October) - 2021" in repr(m4)
+ assert "Master for Month(February) - 2021" in repr(m5)
+ assert "Master for Month(May) - 2030" in repr(m6)
assert m["Chutney Bridge.xlsm"]["Project/Programme Name"] == "Chutney Bridge Ltd"
assert isinstance(m.month, Month)
assert m.month.name == "July"