def test_basic(self): """Test it fills in the points it's meant to.""" plugin = PhaseChangeLevel(phase_change='snow-sleet') plugin.fill_in_sea_points(self.phase_change_level, self.land_sea, self.max_wb_integral, self.wet_bulb_temperature, self.heights) self.assertArrayAlmostEqual(self.phase_change_level.data, self.expected_phase_change_level)
def test_no_sea(self): """Test it only fills in sea points, and ignores a land point""" plugin = PhaseChangeLevel(phase_change='snow-sleet') expected = np.ones((3, 3)) * np.nan land_sea = np.ones((3, 3)) plugin.fill_in_sea_points(self.phase_change_level, land_sea, self.max_wb_integral, self.wet_bulb_temperature, self.heights) self.assertArrayAlmostEqual(self.phase_change_level.data, expected)
def test_all_above_threshold(self): """Test it doesn't change points that are all above the threshold""" plugin = PhaseChangeLevel(phase_change='snow-sleet') self.max_wb_integral[0, 1] = 100 self.phase_change_level[0, 1] = 100 self.expected_phase_change_level[0, 1] = 100 plugin.fill_in_sea_points(self.phase_change_level, self.land_sea, self.max_wb_integral, self.wet_bulb_temperature, self.heights) self.assertArrayAlmostEqual(self.phase_change_level.data, self.expected_phase_change_level)