From d0606872c3fb4712700f34d89e81c7ff55442d31 Mon Sep 17 00:00:00 2001 From: jdebacker Date: Tue, 23 Apr 2024 21:18:13 -0400 Subject: [PATCH] additional test cases --- ccc/tests/test_get_taxcalc_rates.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/ccc/tests/test_get_taxcalc_rates.py b/ccc/tests/test_get_taxcalc_rates.py index 39e602e4..130076f3 100644 --- a/ccc/tests/test_get_taxcalc_rates.py +++ b/ccc/tests/test_get_taxcalc_rates.py @@ -15,13 +15,17 @@ def test_get_calculator_cps(): @pytest.mark.needs_puf @pytest.mark.parametrize( - "data", ["puf.csv", None], ids=["data=PUF", "data=None"] + "baseline,data", + [(True, "puf.csv"), (True, None), (False, None)], + ids=["baseline,data=PUF", "baseline,data=None", "reform,data=None"], ) -def test_get_calculator(data): +def test_get_calculator(baseline, data): """ Test the get_calculator() function """ - calc1 = tc.get_calculator(True, 2019, data=data) + calc1 = tc.get_calculator( + baseline, 2019, baseline_policy={}, reform={}, data=data + ) assert calc1.current_year == 2019