add tests for puf and tmd files

PSLmodels · May 28, 2024 · 62e735e · 62e735e
1 parent 1c4cbef
commit 62e735e
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/taxbrain/tests/test_brain.py b/taxbrain/tests/test_brain.py
@@ -36,6 +36,24 @@ def test_baseline_policy():
     tb.run()
 
 
+@pytest.mark.requires_pufcsv
+def test_baseline_policy_PUF():
+    base = {"II_em": {2019: 0}}
+    reform = {"II_em": {2025: 2000}}
+
+    tb = TaxBrain(2018, 2019, microdata="PUF", reform=reform, base_policy=base)
+    tb.run()
+
+
+@pytest.mark.requires_tmdcsv
+def test_baseline_policy_TMD():
+    base = {"II_em": {2019: 0}}
+    reform = {"II_em": {2025: 2000}}
+
+    tb = TaxBrain(2018, 2019, microdata="TMD", reform=reform, base_policy=base)
+    tb.run()
+
+
 def test_run_corporate_distribution():
     """
     Tests the taxbrain.run method with the distribution of corporate