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)