def test_03_house_duplicate(self):
ri = RoundInfo(1, 'Round1')
ri._houses.append(mhi1) # Prepopulate list of houses
with self.assertRaises(ValueError) as e:
ri.add_house(mhi1)
self.assertEqual(e.exception.args[0],
"House is already present in round")
def test_01_house_type(self):
ri = RoundInfo(1, 'Round1')
oi = OrderInfo(14, 'Road1')
with self.assertRaises(ValueError) as e:
ri.add_house(oi)
self.assertEqual(e.exception.args[0],
"Can only add instances of HouseInfo")
def test_02_house_type(self):
ri = RoundInfo(1, 'Round1')
ri.add_house(mhi1)
self.assertEqual(ri._houses, [mhi1])
def test_01_add_house(self): '''Test the addition of a house to an empty round''' ri = RoundInfo(1, 'Round1') ri.add_house(hi1) self.assertEqual(ri._houses, [hi1])