Add tests

Author: uvchik

tests/test_wake_losses.py

@@ -3,8 +3,9 @@
 import pytest
 from pandas.util.testing import  assert_series_equal
 
-from windpowerlib.wake_losses import reduce_wind_speed
-import windpowerlib.wind_turbine as wt
+from windpowerlib.wake_losses import (reduce_wind_speed,
+                                      get_wind_efficiency_curve)
+
 
 class TestWakeLosses:
 
@@ -30,3 +31,16 @@ def test_reduce_wind_speed(self):
         with pytest.raises(ValueError):
             parameters['wind_efficiency_curve_name'] = 'dena_misspelled'
             reduce_wind_speed(**parameters)
+
+    def test_get_wind_efficiency_curve_one(self):
+        """Test get_wind_efficiency_curve() for one curve."""
+        wec = get_wind_efficiency_curve().sum()
+        wec_exp = pd.Series({'efficiency': 162.45047,
+                             'wind_speed': 1915.23620})
+        assert_series_equal(wec.sort_index(), wec_exp.sort_index())
+
+    def test_get_wind_efficiency_curve_all(self):
+        """Test get_wind_efficiency_curve() for all curves."""
+        wec_all_sum = int(get_wind_efficiency_curve('all').sum().round().sum())
+        print(wec_all_sum)
+        assert wec_all_sum == 12145