Test only for columns that exist in baseline

NREL · Jun 10, 2024 · c9e9fab · c9e9fab
1 parent f5992ba
commit c9e9fab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_supply_curve_aggregation.py b/tests/test_supply_curve_aggregation.py
@@ -56,6 +56,7 @@ def check_agg(agg_out, baseline_h5):
 
                 truth = truth.fillna('none')
                 test = test.fillna('none')
+                test = test[truth.columns]
                 assert_frame_equal(truth, test, check_dtype=False, rtol=0.0001,
                                    check_index_type=False)
             else: