def test_basic(self):
"""Test we fill in the correct snow falling levels for a simple case"""
plugin = PhaseChangeLevel(phase_change='snow-sleet')
plugin.find_extrapolated_falling_level(
self.max_wb_integral, self.gradients, self.intercepts,
self.phase_change_level, self.sea_points)
self.assertArrayAlmostEqual(self.expected_phase_change_level,
self.phase_change_level)
def test_gradients_zero(self):
"""Test we do nothing if all gradients are zero"""
plugin = PhaseChangeLevel(phase_change='snow-sleet')
gradients = np.zeros((3, 3))
plugin.find_extrapolated_falling_level(
self.max_wb_integral, gradients, self.intercepts,
self.phase_change_level, self.sea_points)
expected_phase_change_level = np.ones((3, 3))*np.nan
self.assertArrayAlmostEqual(expected_phase_change_level,
self.phase_change_level)
