Separate temperature tests

Author: uvchik

tests/test_temperature.py

@@ -9,16 +9,18 @@
 class TestTemperature:
 
     def test_linear_gradient(self):
+        """Test temperature as pd.Series"""
         parameters = {'temperature': pd.Series(data=[267, 268]),
                       'temperature_height': 2,
                       'hub_height': 100}
-
-        # Test temperature as pd.Series
         temp_hub_exp = pd.Series(data=[266.363, 267.36300])
         assert_series_equal(linear_gradient(**parameters), temp_hub_exp)
 
-        # Test temperature as np.array
+    def test_temperature_as_np_array(self):
+        """Test temperature as np.array"""
+        parameters = {'temperature': np.array([267, 268]),
+                      'temperature_height': 2,
+                      'hub_height': 100}
         temp_hub_exp = np.array([266.363, 267.36300])
-        parameters['temperature'] = np.array(parameters['temperature'])
         assert_array_equal(linear_gradient(**parameters), temp_hub_exp)
         assert isinstance(linear_gradient(**parameters), np.ndarray)