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)
