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])