Revise test to include RECID in dump output variable list

PSLmodels · Jul 24, 2024 · f272eee · f272eee
1 parent 4e167ba
commit f272eee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/taxcalc/taxcalcio.py b/taxcalc/taxcalcio.py
@@ -519,8 +519,8 @@ def write_output_file(self, output_dump, dump_varset,
             )
             column_order = sorted(outdf.columns)
             # place RECID at start of column_order list
-            if 'RECID' in column_order:
-                column_order.remove('RECID')
+            assert 'RECID' in column_order, 'RECID not in dump output list'
+            column_order.remove('RECID')
             column_order.insert(0, 'RECID')
         else:
             outdf = self.minimal_output()

diff --git a/taxcalc/tests/test_taxcalcio.py b/taxcalc/tests/test_taxcalcio.py
@@ -436,7 +436,7 @@ def test_output_options(reformfile1, assumpfile1):
     # --dump output with partial dump
     try:
         tcio.analyze(writing_output_file=True,
-                     dump_varset=set(['combined']),
+                     dump_varset=set(['RECID', 'combined']),
                      output_dump=True)
     except Exception:  # pylint: disable=broad-except
         if os.path.isfile(outfilepath):