def test_no_fill_if_conditions_not_met(self):
"""Test it doesn't fill in NaN if the heighest wet bulb integral value
is less than the threshold."""
plugin = PhaseChangeLevel(phase_change='snow-sleet')
expected = np.array([[1.0, 1.0, 2.0], [1.0, np.nan, 2.0],
[1.0, 2.0, 2.0]])
plugin.fill_in_high_phase_change_falling_levels(
self.phase_change_level_data, self.orog, self.highest_wb_int,
self.highest_height)
self.assertArrayEqual(self.phase_change_level_data, expected)
def test_basic(self):
"""Test fills in missing data with orography + highest height"""
plugin = PhaseChangeLevel(phase_change='snow-sleet')
self.highest_wb_int[1, 1] = 100.0
expected = np.array([[1.0, 1.0, 2.0], [1.0, 301.0, 2.0],
[1.0, 2.0, 2.0]])
plugin.fill_in_high_phase_change_falling_levels(
self.phase_change_level_data, self.orog, self.highest_wb_int,
self.highest_height)
self.assertArrayEqual(self.phase_change_level_data, expected)
