def test_outside_range(self): """Test method returns an nan if data outside range""" plugin = PhaseChangeLevel(phase_change="snow-sleet") wb_int_data = self.wb_int_data wb_int_data[2, 1, 1] = 70.0 result = plugin.find_falling_level(wb_int_data, self.orog_data, self.height_points) self.assertTrue(np.isnan(result[1, 1]))
def test_basic(self): """Test method returns an array with correct data""" plugin = PhaseChangeLevel(phase_change="snow-sleet") expected = np.array([[10.0, 7.5], [25.0, 20.5]]) result = plugin.find_falling_level(self.wb_int_data, self.orog_data, self.height_points) self.assertIsInstance(result, np.ndarray) self.assertArrayEqual(result, expected)