Add test for np.array to test_smallest_difference()

Author: SabineHaas

tests/test_tools.py

@@ -1,6 +1,8 @@
 from windpowerlib.tools import smallest_difference, linear_extra_interpolation
 import pandas as pd
+import numpy as np
 from pandas.util.testing import assert_series_equal
+from numpy.testing import assert_array_equal, assert_allclose
 
 
 class TestTools:
@@ -12,37 +14,52 @@ def test_smallest_difference(self):
             pd.Series(data=[4.0, 5.0, 6.0]),
             pd.Series(data=[8.0, 10.0, 14.0]),
             pd.Series(data=[16.0, 20.0, 28.0])]}, index=[100, 150, 200])
+        data_frame_arr = pd.DataFrame(data={'v_wind': [
+            np.array([4.0, 5.0, 6.0]),
+            np.array([8.0, 10.0, 14.0]),
+            np.array([16.0, 20.0, 28.0])]}, index=[100, 150, 200])
         parameters = {'comp_value': 100,
                       'column_name': 'v_wind'}
-        # TODO: test v_wind as np.array
 
         # comparative value is an index of data frame
         exp_output = (100, 4.0)
         assert smallest_difference(data_frame, **parameters) == exp_output
         exp_series = pd.Series(data=[4.0, 5.0, 6.0])
         assert_series_equal(smallest_difference(data_frame_series,
                                                 **parameters)[1], exp_series)
+        exp_arr = np.array([4.0, 5.0, 6.0])
+        assert_array_equal(smallest_difference(data_frame_arr,
+                                               **parameters)[1], exp_arr)
         # comparative value between indices of data frame
         exp_output = (150, 5.0)
         parameters['comp_value'] = 175
         assert smallest_difference(data_frame, **parameters) == exp_output
         exp_series = pd.Series(data=[8.0, 10.0, 14.0])
         assert_series_equal(smallest_difference(data_frame_series,
                                                 **parameters)[1], exp_series)
+        exp_arr = np.array([8.0, 10.0, 14.0])
+        assert_array_equal(smallest_difference(data_frame_arr,
+                                               **parameters)[1], exp_arr)
         # comparative value > indices of data frame
         exp_output = (200, 6.0)
         parameters['comp_value'] = 250
         assert smallest_difference(data_frame, **parameters) == exp_output
         exp_series = pd.Series(data=[16.0, 20.0, 28.0])
         assert_series_equal(smallest_difference(data_frame_series,
                                                 **parameters)[1], exp_series)
+        exp_arr = np.array([16.0, 20.0, 28.0])
+        assert_array_equal(smallest_difference(data_frame_arr,
+                                               **parameters)[1], exp_arr)
         # comparative value < indices of data frame
         exp_output = (100, 4.0)
         parameters['comp_value'] = 90
         assert smallest_difference(data_frame, **parameters) == exp_output
         exp_series = pd.Series(data=[4.0, 5.0, 6.0])
         assert_series_equal(smallest_difference(data_frame_series,
                                                 **parameters)[1], exp_series)
+        exp_arr = np.array([4.0, 5.0, 6.0])
+        assert_array_equal(smallest_difference(data_frame_arr,
+                                               **parameters)[1], exp_arr)
 
     def test_linear_extra_interpolation(self):
         weather = pd.DataFrame(data={'v_wind': [4.0, 5.0, 6.0]},