diff --git a/make_post_sell/tests/test_models.py b/make_post_sell/tests/test_models.py index 60cc6ff..599cbd4 100644 --- a/make_post_sell/tests/test_models.py +++ b/make_post_sell/tests/test_models.py @@ -3238,12 +3238,14 @@ class TestAnalyticsHelpers(unittest.TestCase): from ..views.analytics import _fmt_score self.assertEqual(_fmt_score(0.0), "0.0") - def test_cutoffs_are_7_14_21_days_apart(self): + def test_cutoffs_returns_expected_keys(self): from ..views.analytics import _cutoffs - c7, c14, c21 = _cutoffs() + cuts = _cutoffs() day_ms = 24 * 60 * 60 * 1000 - self.assertAlmostEqual(c7 - c14, 7 * day_ms, delta=1000) - self.assertAlmostEqual(c14 - c21, 7 * day_ms, delta=1000) + for key in ("1d", "7d", "14d", "21d", "28d", "365d"): + self.assertIn(key, cuts) + self.assertAlmostEqual(cuts["7d"] - cuts["14d"], 7 * day_ms, delta=1000) + self.assertAlmostEqual(cuts["14d"] - cuts["21d"], 7 * day_ms, delta=1000) def test_referrer_labels_complete(self): from ..views.analytics import REFERRER_LABELS