refactoring the test, using pytest

Author: aman-coder03

tests/test_transformer.py

@@ -4,6 +4,7 @@
 
 from pvlib import transformer
 
+import pytest
 
 def test_simple_efficiency():
 
@@ -59,12 +60,18 @@ def test_simple_efficiency_known_values():
         rating,
     )
 
+@pytest.mark.parametrize(
+    "input_power, no_load_loss, load_loss, transformer_rating, expected",
+    [
+        (1000.0, 0.01, 0.0, 1000.0, 990.0),
+    ],
+)
 
-def test_simple_efficiency_zero_load_loss():
-    input_power = 1000.0
-    no_load_loss = 0.01
-    load_loss = 0.0
-    transformer_rating = 1000.0
+def test_simple_efficiency_zero_load_loss(
+    input_power, no_load_loss, load_loss, transformer_rating, expected
+):
+    # for load_loss = 0, the model reduces to:
+    # P_out = P_in - L_no_load * P_nom
 
     result = transformer.simple_efficiency(
         input_power=input_power,
@@ -73,8 +80,4 @@ def test_simple_efficiency_zero_load_loss():
         transformer_rating=transformer_rating,
     )
 
-    # for load_loss = 0, the model reduces to:
-    # P_out = P_in - L_no_load * P_nom
-    expected = input_power - no_load_loss * transformer_rating
-
     assert_allclose(result, expected)