Don't use enum in pytest parametrize pt 2

NREL · May 29, 2024 · 36d00ef · 36d00ef
1 parent ce68fec
commit 36d00ef
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_handlers_transmission.py b/tests/test_handlers_transmission.py
@@ -116,8 +116,7 @@ def test_cost_calculation(i, trans_costs, distance, gid, trans_table):
     assert true_cost == trans_cost
 
 
-@pytest.mark.parametrize(('trans_costs', SupplyCurveField.CAPACITY,
-                          SupplyCurveField.GID),
+@pytest.mark.parametrize(('trans_costs', "capacity", "gid"),
                          ((TRANS_COSTS_1, 350, 43300),
                           (TRANS_COSTS_2, 350, 43300),
                           (TRANS_COSTS_1, 100, 43300),