Refactor pyomo handling in stand_by_cost test

PyPSA · Dec 3, 2023 · ff10e4a · ff10e4a
1 parent c5b10c7
commit ff10e4a
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/test/test_stand_by_cost.py b/test/test_stand_by_cost.py
@@ -2,20 +2,22 @@
 import numpy as np
 import pandas as pd
 import pytest
-from conftest import optimize
+from conftest import optimize, SUPPORTED_APIS
 from numpy.testing import assert_array_almost_equal as equal
 
 import pypsa
 
-SUPPORTED_APIS_STAND_BY_COST = ["linopy", "native"]
 
-
-@pytest.mark.parametrize("api", SUPPORTED_APIS_STAND_BY_COST)
+@pytest.mark.parametrize("api", SUPPORTED_APIS)
 def test_stand_by_cost(api):
     """
     This test is based on https://pypsa.readthedocs.io/en/latest/examples/unit-
     commitment.html and is not very comprehensive.
     """
+
+    if api == "pyomo":
+        pytest.skip("stand_by_cost not implemented for pyomo")
+
     n = pypsa.Network()
 
     snapshots = range(4)