def test_remove_measure_pass(self): """Test remove_measure removes Measure of MeasureSet correcty.""" meas = MeasureSet() act_1 = Measure() act_1.name = 'Mangrove' act_1.haz_type = 'FL' meas.append(act_1) meas.remove_measure(name='Mangrove') self.assertEqual(0, meas.size())
def test_remove_wrong_error(self): """Test error is raised when invalid inputs.""" meas = MeasureSet() act_1 = Measure() act_1.name = 'Mangrove' act_1.haz_type = 'FL' meas.append(act_1) with self.assertLogs('climada.entity.measures.measure_set', level='INFO') as cm: meas.remove_measure(name='Seawall') self.assertIn('No Measure with name Seawall.', cm.output[0])