Split long test method to make it easier to find failing tests

Author: uvchik

tests/test_power_output.py

@@ -25,10 +25,25 @@ def setup_class(self):
             "power_coefficient_curve_wind_speeds": pd.Series([4.0, 5.0, 6.0]),
             "power_coefficient_curve_values": pd.Series([0.3, 0.4, 0.5]),
         }
+        self.parameters2: Dict = {
+            "wind_speed": pd.Series(data=[2.0, 5.5, 7.0]),
+            "density": pd.Series(data=[1.3, 1.3, 1.3]),
+            "density_correction": False,
+            "power_curve_wind_speeds": pd.Series([4.0, 5.0, 6.0]),
+            "power_curve_values": pd.Series([300, 400, 500]),
+        }
+        self.power_output_exp1 = pd.Series(
+            data=[0.0, 450.0, 0.0], name="feedin_power_plant"
+        )
+        self.power_output_exp2 = pd.Series(
+            data=[0.0, 461.00290572, 0.0], name="feedin_power_plant"
+        )
 
     def test_power_coefficient_curve_1(self):
-        # Test wind_speed as pd.Series with density and power_coefficient_curve
-        # as pd.Series and np.array
+        """
+        Test wind_speed as pd.Series with density and power_coefficient_curve
+        as pd.Series and np.array
+        """
         power_output_exp = pd.Series(
             data=[0.0, 244615.399, 0.0], name="feedin_power_plant"
         )
@@ -53,14 +68,20 @@ def test_power_coefficient_curve_1(self):
         )
 
     def test_power_coefficient_curve_output_types(self):
-        parameters = self.parameters
-        # Test wind_speed as np.array with density and power_coefficient_curve
-        # as np.array and pd.Series
-        assert isinstance(power_coefficient_curve(**parameters), pd.Series)
-        parameters["wind_speed"] = np.array(parameters["wind_speed"])
-        assert isinstance(power_coefficient_curve(**parameters), np.ndarray)
+        """
+        Test wind_speed as np.array with density and power_coefficient_curve
+        as np.array and pd.Series
+        """
+        assert isinstance(
+            power_coefficient_curve(**self.parameters), pd.Series
+        )
+        self.parameters["wind_speed"] = np.array(self.parameters["wind_speed"])
+        assert isinstance(
+            power_coefficient_curve(**self.parameters), np.ndarray
+        )
 
     def test_power_coefficient_curve_2(self):
+        """TODO: Explain this test"""
         parameters = self.parameters
         power_output_exp = np.array([0.0, 244615.399, 0.0])
         parameters["wind_speed"] = np.array(parameters["wind_speed"])
@@ -83,86 +104,124 @@ def test_power_coefficient_curve_2(self):
         )
         assert isinstance(power_coefficient_curve(**parameters), np.ndarray)
 
-    def test_power_curve(self):
-        parameters = {
-            "wind_speed": pd.Series(data=[2.0, 5.5, 7.0]),
-            "density": pd.Series(data=[1.3, 1.3, 1.3]),
-            "density_correction": False,
-            "power_curve_wind_speeds": pd.Series([4.0, 5.0, 6.0]),
-            "power_curve_values": pd.Series([300, 400, 500]),
-        }
-
+    def test_power_curve_1(self):
         # Tests without density correction:
         # Test wind_speed as pd.Series and power_curve as pd.Series and
         # np.array
-        power_output_exp = pd.Series(
-            data=[0.0, 450.0, 0.0], name="feedin_power_plant"
+
+        assert_series_equal(
+            power_curve(**self.parameters2), self.power_output_exp1
         )
-        assert_series_equal(power_curve(**parameters), power_output_exp)
-        parameters["power_curve_values"] = np.array(
-            parameters["power_curve_values"]
+
+    def test_power_curve_2(self):
+        """TODO: Explain this test"""
+        self.parameters2["power_curve_values"] = np.array(
+            self.parameters2["power_curve_values"]
         )
-        parameters["power_curve_wind_speeds"] = np.array(
-            parameters["power_curve_wind_speeds"]
+        self.parameters2["power_curve_wind_speeds"] = np.array(
+            self.parameters2["power_curve_wind_speeds"]
+        )
+        assert_series_equal(
+            power_curve(**self.parameters2), self.power_output_exp1
         )
-        assert_series_equal(power_curve(**parameters), power_output_exp)
 
-        # Test wind_speed as np.array and power_curve as pd.Series and np.array
-        power_output_exp = np.array([0.0, 450.0, 0.0])
-        parameters["wind_speed"] = np.array(parameters["wind_speed"])
-        assert_allclose(power_curve(**parameters), power_output_exp)
-        assert isinstance(power_curve(**parameters), np.ndarray)
-        parameters["power_curve_wind_speeds"] = pd.Series(
-            data=parameters["power_curve_wind_speeds"]
+    def test_power_curve_3(self):
+        """
+        Test wind_speed as np.array and power_curve as
+        pd.Series and np.array
+        """
+        power_output_exp = np.array(self.power_output_exp1)
+        self.parameters2["wind_speed"] = np.array(
+            self.parameters2["wind_speed"]
+        )
+        assert_allclose(power_curve(**self.parameters2), power_output_exp)
+        assert isinstance(power_curve(**self.parameters2), np.ndarray)
+
+    def test_power_curve_4(self):
+        """TODO: Explain this test"""
+        self.parameters2["power_curve_wind_speeds"] = pd.Series(
+            data=self.parameters2["power_curve_wind_speeds"]
         )
-        parameters["power_curve_values"] = pd.Series(
-            data=parameters["power_curve_values"]
+        self.parameters2["power_curve_values"] = pd.Series(
+            data=self.parameters2["power_curve_values"]
         )
-        assert_allclose(power_curve(**parameters), power_output_exp)
-        assert isinstance(power_curve(**parameters), np.ndarray)
+        assert_allclose(
+            power_curve(**self.parameters2), self.power_output_exp1
+        )
+        assert isinstance(power_curve(**self.parameters2), np.ndarray)
 
-        # Tests with density correction:
-        # Test wind_speed as np.array with density and power_curve as pd.Series
-        # and np.array
-        power_output_exp = np.array([0.0, 461.00290572, 0.0])
-        parameters["density_correction"] = True
-        assert_allclose(power_curve(**parameters), power_output_exp)
-        assert isinstance(power_curve(**parameters), np.ndarray)
-        parameters["density"] = np.array(parameters["density"])
-        assert_allclose(power_curve(**parameters), power_output_exp)
-        assert isinstance(power_curve(**parameters), np.ndarray)
-        parameters["power_curve_values"] = np.array(
-            parameters["power_curve_values"]
+    def test_power_curve_5(self):
+        """
+        Tests with density correction:
+        Test wind_speed as np.array with density and power_curve as pd.Series
+        and np.array
+        """
+        power_output_exp = np.array(self.power_output_exp2)
+        self.parameters2["density_correction"] = True
+        assert_allclose(power_curve(**self.parameters2), power_output_exp)
+        assert isinstance(power_curve(**self.parameters2), np.ndarray)
+
+    def test_power_curve_6(self):
+        """TODO: Explain this test"""
+        self.parameters2["density"] = np.array(self.parameters2["density"])
+        assert_allclose(
+            power_curve(**self.parameters2), self.power_output_exp2
         )
-        parameters["power_curve_wind_speeds"] = np.array(
-            parameters["power_curve_wind_speeds"]
+        assert isinstance(power_curve(**self.parameters2), np.ndarray)
+
+    def test_power_curve_7(self):
+        """TODO: Explain this test"""
+        self.parameters2["power_curve_values"] = np.array(
+            self.parameters2["power_curve_values"]
         )
-        assert_allclose(power_curve(**parameters), power_output_exp)
-        assert isinstance(power_curve(**parameters), np.ndarray)
+        self.parameters2["power_curve_wind_speeds"] = np.array(
+            self.parameters2["power_curve_wind_speeds"]
+        )
+        assert_allclose(
+            power_curve(**self.parameters2), self.power_output_exp2
+        )
+        assert isinstance(power_curve(**self.parameters2), np.ndarray)
 
-        # Test wind_speed as pd.Series with density and power_curve as
-        # np. array and pd.Series
-        power_output_exp = pd.Series(
-            data=[0.0, 461.00290572, 0.0], name="feedin_power_plant"
+    def test_power_curve_8(self):
+        """
+        Test wind_speed as pd.Series with density and power_curve as np. array
+         and pd.Series
+        """
+        self.parameters2["wind_speed"] = pd.Series(
+            data=self.parameters2["wind_speed"]
         )
-        parameters["wind_speed"] = pd.Series(data=parameters["wind_speed"])
-        assert_series_equal(power_curve(**parameters), power_output_exp)
-        parameters["density"] = pd.Series(data=parameters["density"])
-        assert_series_equal(power_curve(**parameters), power_output_exp)
-        parameters["power_curve_wind_speeds"] = pd.Series(
-            data=parameters["power_curve_wind_speeds"]
+        assert_series_equal(
+            power_curve(**self.parameters2), self.power_output_exp2
         )
-        parameters["power_curve_values"] = pd.Series(
-            data=parameters["power_curve_values"]
+
+    def test_power_curve_9(self):
+        """TODO: Explain this test"""
+        self.parameters2["density"] = pd.Series(
+            data=self.parameters2["density"]
+        )
+        assert_series_equal(
+            power_curve(**self.parameters2), self.power_output_exp2
+        )
+
+    def test_power_curve_10(self):
+        """TODO: Explain this test"""
+        self.parameters2["power_curve_wind_speeds"] = pd.Series(
+            data=self.parameters2["power_curve_wind_speeds"]
+        )
+        self.parameters2["power_curve_values"] = pd.Series(
+            data=self.parameters2["power_curve_values"]
+        )
+        assert_series_equal(
+            power_curve(**self.parameters2), self.power_output_exp2
         )
-        assert_series_equal(power_curve(**parameters), power_output_exp)
 
         # Raise TypeErrors due to wrong type of `density_correction`
         with pytest.raises(TypeError):
             parameters["density"] = "wrong_type"
             power_curve(**parameters)
 
     def test_power_curve_density_correction(self):
+        """TODO: Explain and split this test."""
         parameters = {
             "wind_speed": pd.Series(data=[2.0, 5.5, 7.0]),
             "density": pd.Series(data=[1.3, 1.3, 1.3]),