Test revised WindTurbine class

Author: uvchik

tests/test_wind_turbine.py

@@ -1,28 +1,58 @@
+"""
+Testing the wind_turbine module.
+"""
+
+__copyright__ = "Copyright oemof developer group"
+__license__ = "GPLv3"
+
 import pytest
 import os
+from windpowerlib.tools import WindpowerlibUserWarning
 
 from windpowerlib.wind_turbine import (get_turbine_data_from_file, WindTurbine,
-                                       get_turbine_types)
+                                       get_turbine_types,
+                                       load_turbine_data_from_oedb)
 
 
 class TestWindTurbine:
 
-    def test_error_raising(self):
+    def test_warning(self, recwarn):
         source = os.path.join(os.path.dirname(__file__), '../example/data')
         self.test_turbine_data = {'hub_height': 100,
                                   'rotor_diameter': 80,
                                   'turbine_type': 'turbine_not_in_file',
                                   'path': source}
-        # Raise system exit due to turbine type not in file
-        # with pytest.raises(SystemExit):
         assert(WindTurbine(**self.test_turbine_data).power_curve is None)
+        assert recwarn.pop(WindpowerlibUserWarning)
 
     def test_get_turbine_data_from_file(self):
         # Raise FileNotFoundError due to missing
         with pytest.raises(FileNotFoundError):
             get_turbine_data_from_file(turbine_type='...',
                                        path='not_existent')
 
-    def test_get_turbine_types(self):
-        get_turbine_types(print_out=True, filter_=True)
-        get_turbine_types(print_out=False, filter_=False)
+    def test_get_turbine_types(self, capsys):
+        get_turbine_types()
+        captured = capsys.readouterr()
+        assert 'Enercon' in captured.out
+        get_turbine_types('oedb', print_out=False, filter_=False)
+        msg = "`turbine_library` is 'wrong' but must be 'local' or 'oedb'."
+        with pytest.raises(ValueError, match=msg):
+            get_turbine_types('wrong')
+
+    def test_deduce_nominal_power(self):
+        """Test method to deduce nominal_power from power curve"""
+        test_turbine_data = {'hub_height': 100,
+                             'rotor_diameter': 80,
+                             'turbine_type': 'N131/3000'}
+        n131 = WindTurbine(**test_turbine_data)
+        assert n131.nominal_power == 3000000.0
+        n131.deduce_nominal_power_from_power_curve()
+        assert n131.nominal_power == 3000000.0
+
+    def test_wrong_url_load_turbine_data(self):
+        """Load turbine data from oedb."""
+
+        with pytest.raises(ConnectionError,
+                           match="Database connection not successful"):
+            load_turbine_data_from_oedb('wrong_schema')

windpowerlib/wind_turbine.py

@@ -259,28 +259,31 @@ def get_turbine_data_from_file(turbine_type, path):
         return wpp_df
 
 
-def load_turbine_data_from_oedb():
+def load_turbine_data_from_oedb(schema = 'supply', table = 'turbine_library'):
     r"""
     Loads turbine library from the OpenEnergy database (oedb).
 
     Turbine data is saved to csv files ('oedb_power_curves.csv',
     'oedb_power_coefficient_curves.csv' and 'oedb_nominal_power') for offline
     usage of the windpowerlib. If the files already exist they are overwritten.
 
+    Parameters
+    ----------
+    schema : str
+        Database schema of the turbine library.
+    table : str
+        Table name of the turbine library.
+
     Returns
     -------
     :pandas:`pandas.DataFrame<frame>`
         Turbine data of different turbines such as 'manufacturer',
         'turbine_type', 'nominal_power'.
 
     """
-    # TODO: Test is missing
-
     # url of OpenEnergy Platform that contains the oedb
     oep_url = 'http://oep.iks.cs.ovgu.de/'
-    # location of data
-    schema = 'supply'
-    table = 'turbine_library'
+
     # load data
     result = requests.get(
         oep_url + '/api/v0/schema/{}/tables/{}/rows/?'.format(
@@ -399,7 +402,7 @@ def get_turbine_types(turbine_library='local', print_out=True, filter_=True):
     elif turbine_library == 'oedb':
         df = load_turbine_data_from_oedb()
     else:
-        raise ValueError("`turbine_library` is {} ".format(turbine_library) +
+        raise ValueError("`turbine_library` is '{}' ".format(turbine_library) +
                          "but must be 'local' or 'oedb'.")
     if filter_:
         cp_curves_df = df.loc[df['has_cp_curve']][