Add tests

Author: uvchik

tests/test_wind_turbine.py

@@ -16,12 +16,16 @@
 
 class TestWindTurbine:
 
+    @classmethod
+    def setup_class(cls):
+        """Setup default values"""
+        cls.source = os.path.join(os.path.dirname(__file__), '../example/data')
+
     def test_warning(self, recwarn):
-        source = os.path.join(os.path.dirname(__file__), '../example/data')
         test_turbine_data = {'hub_height': 100,
                              'rotor_diameter': 80,
                              'turbine_type': 'turbine_not_in_file',
-                             'path': source}
+                             'path': self.source}
         assert(WindTurbine(**test_turbine_data).power_curve is None)
         assert recwarn.pop(WindpowerlibUserWarning)
 
@@ -50,3 +54,27 @@ def test_wrong_url_load_turbine_data(self):
     @pytest.mark.filterwarnings("ignore:The WindTurbine")
     def test_string_representation_of_wind_turbine(self):
         assert "Wind turbine: ['hub height=120 m'" in repr(WindTurbine(120))
+
+    def test_to_group_method(self):
+        example_turbine = {
+           'hub_height': 100,
+           'rotor_diameter': 70,
+           'turbine_type': 'DUMMY 3',
+           'path': self.source}
+        e_t_1 = WindTurbine(**example_turbine)
+        assert(isinstance(e_t_1.to_group(5), dict))
+        assert(e_t_1.to_group(5)['number_of_turbines'] == 5)
+        assert(e_t_1.to_group(number_turbines=5)['number_of_turbines'] == 5)
+        assert(e_t_1.to_group(total_capacity=3e6)['number_of_turbines'] == 2.0)
+
+    def test_wrongly_defined_to_group_method(self):
+        example_turbine = {
+           'hub_height': 100,
+           'rotor_diameter': 70,
+           'turbine_type': 'DUMMY 3',
+           'path': self.source}
+        e_t_1 = WindTurbine(**example_turbine)
+        with pytest.raises(ValueError,
+                           match="The 'number' and the 'total_capacity"
+                                 " parameter are mutually exclusive."):
+            e_t_1.to_group(5, 3000)

windpowerlib/wind_farm.py

@@ -93,6 +93,12 @@ class WindFarm(object):
     >>> example_farm = wind_farm.WindFarm(**example_farm_data)
     >>> print(example_farm.nominal_power)
     31200000.0
+    >>> # turbine fleet as list of WindTurbine using the 'to_group' method.
+    >>> wind_turbine_fleet = [e126.to_group(number_turbines=6),
+    ...                       v90.to_group(total_capacity=3 * 2e6)]
+    >>> example_farm = wind_farm.WindFarm(wind_turbine_fleet)
+    >>> print(example_farm.nominal_power)
+    31200000.0
 
     """
 

windpowerlib/wind_turbine.py

@@ -200,8 +200,8 @@ def __repr__(self):
 
     def to_group(self, number_turbines=None, total_capacity=None):
         if number_turbines is not None and total_capacity is not None:
-            raise ValueError("The 'number' and the 'total_capacity are"
-                             " mutually exclusive. Use just one of them.")
+            raise ValueError("The 'number' and the 'total_capacity parameter "
+                             "are mutually exclusive. Use just one of them.")
         elif total_capacity is not None:
             number_turbines = total_capacity / self.nominal_power
         elif number_turbines is None: