def test_remove_add_pass(self):
"""Test ImpactFunc can be added after removing."""
imp_fun = ImpactFuncSet()
vulner_1 = ImpactFunc()
vulner_1.id = 1
vulner_1.haz_type = 'TC'
imp_fun.append(vulner_1)
imp_fun.remove_func()
self.assertEqual(0, len(imp_fun.get_hazard_types()))
self.assertEqual(0, len(imp_fun.get_ids()))
imp_fun.append(vulner_1)
self.assertEqual(1, len(imp_fun.get_hazard_types()))
self.assertEqual('TC', imp_fun.get_hazard_types()[0])
self.assertEqual(1, len(imp_fun.get_ids()))
self.assertEqual([1], imp_fun.get_ids('TC'))
def test_get_hazards_pass(self):
"""Test get_hazard_types function."""
imp_fun = ImpactFuncSet()
vulner_1 = ImpactFunc()
vulner_1.id = 1
vulner_1.haz_type = 'TC'
imp_fun.append(vulner_1)
self.assertEqual(1, len(imp_fun.get_hazard_types()))
self.assertEqual(['TC'], imp_fun.get_hazard_types())
vulner_2 = ImpactFunc()
vulner_2.id = 1
vulner_2.haz_type = 'TC'
imp_fun.append(vulner_2)
self.assertEqual(1, len(imp_fun.get_hazard_types()))
self.assertEqual(['TC'], imp_fun.get_hazard_types())
vulner_3 = ImpactFunc()
vulner_3.id = 1
vulner_3.haz_type = 'FL'
imp_fun.append(vulner_3)
self.assertEqual(2, len(imp_fun.get_hazard_types()))
self.assertIn('TC', imp_fun.get_hazard_types())
self.assertIn('FL', imp_fun.get_hazard_types())
