1 files changed, 52 insertions, 0 deletions
diff --git a/datamaps/tests/test_quarter.py b/datamaps/tests/test_quarter.py
new file mode 100644
index 0000000..c5b1824
--- /dev/null
+++ b/datamaps/tests/test_quarter.py
@@ -0,0 +1,52 @@
+import datetime
+
+import pytest
+
+from ..core import Quarter
+
+
+def test_initialisation():
+    q = Quarter(1, 2017)
+    assert q.start_date == datetime.date(2017, 4, 1)
+    assert q.end_date == datetime.date(2017, 6, 30)
+    q = Quarter(2, 2017)
+    assert q.start_date == datetime.date(2017, 7, 1)
+    assert q.end_date == datetime.date(2017, 9, 30)
+    q = Quarter(4, 2017)
+    assert q.start_date == datetime.date(2018, 1, 1)
+    assert q.end_date == datetime.date(2018, 3, 31)
+
+
+def test_desc_string():
+    assert str(Quarter(1, 2013)) == "Q1 13/14"
+    assert str(Quarter(2, 2013)) == "Q2 13/14"
+    assert str(Quarter(3, 2013)) == "Q3 13/14"
+    assert str(Quarter(4, 2013)) == "Q4 13/14"
+
+    assert str(Quarter(1, 1998)) == "Q1 98/99"
+    assert str(Quarter(2, 1998)) == "Q2 98/99"
+    assert str(Quarter(3, 1998)) == "Q3 98/99"
+    assert str(Quarter(4, 1998)) == "Q4 98/99"
+
+    assert str(Quarter(1, 1999)) == "Q1 99/00"
+    assert str(Quarter(2, 1999)) == "Q2 99/00"
+    assert str(Quarter(3, 1999)) == "Q3 99/00"
+    assert str(Quarter(4, 1999)) == "Q4 99/00"
+
+
+def test_errors():
+    with pytest.raises(ValueError) as excinfo:
+        Quarter(5, 2017)
+    assert "A quarter must be either 1, 2, 3 or 4" in str(excinfo.value)
+
+    with pytest.raises(ValueError) as excinfo:
+        Quarter(3, 1921)
+    assert "Year must be between 1950 and 2100 - surely that will do?" in str(excinfo.value)
+
+    with pytest.raises(ValueError) as excinfo:
+        Quarter("3", 2016)
+    assert "A quarter must be either 1, 2, 3 or 4" in str(excinfo.value)
+
+    with pytest.raises(ValueError) as excinfo:
+        Quarter(3, "1921")
+    assert "Year must be between 1950 and 2100 - surely that will do?" in str(excinfo.value)