Add weekday.py tests

The expected values in these tests were generated by running the version of these methods on master

Author: alexcoda

_delphi_utils_python/tests/test_weekday.py

@@ -0,0 +1,74 @@
+import logging
+
+import numpy as np
+import pandas as pd
+from delphi_utils.weekday import Weekday
+
+
+class TestWeekday:
+
+    TEST_DATA = pd.DataFrame({
+        "num": np.arange(1, 11, 1),
+        "den": np.arange(11, 21, 1),
+        "date": pd.date_range("2020-01-01", "2020-01-10")
+    })
+
+    def test_get_params(self):
+        TEST_LOGGER = logging.getLogger()
+
+        result = Weekday.get_params(self.TEST_DATA, "den", ["num"], "date", [1], TEST_LOGGER)
+        print(result)
+        expected_result = [
+            -0.05993665,
+            -0.0727396,
+            -0.05618517,
+            0.0343405,
+            0.12534997,
+            0.04561813,
+            -2.27669028,
+            -1.89564374,
+            -1.5695407,
+            -1.29838116,
+            -1.08216513,
+            -0.92089259,
+            -0.81456355,
+            -0.76317802,
+            -0.76673598,
+            -0.82523745,
+        ]
+        assert np.allclose(result, expected_result)
+
+    def test_calc_adjustment_with_zero_parameters(self):
+        params = np.array([[0, 0, 0, 0, 0, 0, 0]])
+
+        result = Weekday.calc_adjustment(params, self.TEST_DATA, ["num"], "date")
+
+        # Data should be unchanged when params are 0's
+        assert np.allclose(result["num"].values, self.TEST_DATA["num"].values)
+        assert np.allclose(result["den"].values, self.TEST_DATA["den"].values)
+        assert np.array_equal(result["date"].values, self.TEST_DATA["date"].values)
+
+    def test_calc_adjustment(self):
+        params = np.array([[1, -1, 1, -1, 1, -1, 1]])
+
+        result = Weekday.calc_adjustment(params, self.TEST_DATA, ["num"], "date")
+
+        print(result["num"].values)
+        print(result["den"].values)
+        expected_nums = [
+            0.36787944,
+            5.43656366,
+            1.10363832,
+            10.87312731,
+            5,
+            2.20727665,
+            19.0279728,
+            2.94303553,
+            24.46453646,
+            3.67879441,
+        ]
+
+        # The date and "den" column are unchanged by this function
+        assert np.allclose(result["num"].values, expected_nums)
+        assert np.allclose(result["den"].values, self.TEST_DATA["den"].values)
+        assert np.array_equal(result["date"].values, self.TEST_DATA["date"].values)