def test_04_house_remove(self):
ri = RoundInfo(1, 'Round1', [mhi1])
oi = OrderInfo(14, 'Road1')
with self.assertRaises(ValueError) as e:
ri.rem_house(oi)
self.assertEqual(e.exception.args[0],
"Can only remove instances of HouseInfo")
def test_09_house_remove(self):
ri = RoundInfo(1, 'Round1', [mhi1, mhi2])
ri.rem_house(mhi1)
ri.rem_house(mhi2)
self.assertEqual(ri._houses, [])
with self.assertRaises(ValueError) as e:
ri.rem_house(mhi1)
self.assertEqual(e.exception.args[0], "House not present in round")
def test_08_house_remove(self):
ri = RoundInfo(1, 'Round1', [mhi1, mhi2])
ri.rem_house(mhi1)
self.assertEqual(ri._houses, [mhi2])