def test_can_upgrade_no_houses(self): bi = Board(["red", "blue"], 10000) bi.available_houses = 1 bi.purchase("red", 1) bi.purchase("red", 3) bi.upgrade("red", 1) self.assertFalse(bi.can_upgrade("red", 1)) self.assertFalse(bi.can_upgrade("red", 3))
def test_can_downgrade_4_houses(self): bi = Board(["red", "blue"], 10000) bi.purchase("red", 1) bi.purchase("red", 3) bi.available_houses = 4 bi.upgrade("red", 1) bi.upgrade("red", 1) bi.upgrade("red", 1) bi.upgrade("red", 1) bi.upgrade("red", 1) self.assertTrue(bi.can_downgrade("red", 1)) self.assertFalse(bi.can_downgrade("blue", 1))