coverage

cumulus_library/statistics/counts/__init__.py

@@ -1 +1 @@
-from cumulus_library.builders.counts import CountsBuilder  # pramga: no cover
+from cumulus_library.builders.counts import CountsBuilder

tests/test_counts_builder.py

@@ -24,6 +24,21 @@ def test_get_table_name(name, duration, expected):
     assert output == expected
 
 
+@pytest.mark.parametrize(
+    "name,duration,expected",
+    [
+        ("table", None, "test__table"),
+        ("table", "month", "test__table_month"),
+    ],
+)
+def test_deprecation_patch(name, duration, expected):
+    from cumulus_library.statistics.counts import CountsBuilder
+
+    builder = CountsBuilder(study_prefix=TEST_PREFIX)
+    output = builder.get_table_name(name, duration)
+    assert output == expected
+
+
 @pytest.mark.parametrize(
     "clause,min_subject,expected,raises",
     [