class BoardTest(unittest.TestCase):
def setUp(self):
self.board = Board()
self.redPlayer = Player(PlayerColor.Red, StupidBot())
self.bluePlayer = Player(PlayerColor.Blue, StupidBot())
self.players = [self.redPlayer, self.bluePlayer]
def testBoardInitialization(self):
self.assertListEqual([7,7,7,7], self.board.numberOfHutsLeft())
self.assertEqual(36, self.board.numberOfCardsLeft())
def testAvailableHuts(self):
ahs = self.board.availableHuts()
self.assertEqual(4, len(ahs))
self.assertIsInstance(ahs[0], Hut)
def testPlaceOnHut(self):
ahs = self.board.availableHuts()
targetHut = ahs[0]
self.board.placeOnHutIndex(0, self.redPlayer)
ahs = self.board.availableHuts()
self.assertEqual(3, len(ahs), "should only be 3 huts left")
self.assertNotIn(targetHut, ahs, "hut should not be available")
def testPersonCountAfterPlacingOnHut(self):
self.assertEqual(0, self.board.person(self.redPlayer))
self.assertEqual(0, self.board.personsOnHuts(self.redPlayer))
self.board.placeOnHutIndex(0, self.redPlayer)
self.assertEqual(1, self.board.person(self.redPlayer))
self.assertEqual(1, self.board.personsOnHuts(self.redPlayer))
self.assertEqual(0, self.board.person(self.bluePlayer))
self.assertEqual(0, self.board.personsOnHuts(self.bluePlayer))
self.board.placeOnHutIndex(1, self.bluePlayer)
self.assertEqual(1, self.board.person(self.bluePlayer))
self.assertEqual(1, self.board.personsOnHuts(self.bluePlayer))
def testPlacePersonsWithoutResources(self):
self.assertEqual(0, self.board.person(self.redPlayer))
self.board.addHunters(2, self.redPlayer)
self.board.addLumberjacks(2, self.redPlayer)
self.assertEqual(4, self.board.person(self.redPlayer))
self.board.addClayDiggers(1, self.redPlayer)
self.assertEqual(5, self.board.person(self.redPlayer))
def testIllegalPlacement(self):
self.board.addStoneDiggers(2, self.redPlayer)
self.assertEqual(2, self.board.person(self.redPlayer))
from Board import PlacementError
with self.assertRaises(PlacementError):
self.board.addStoneDiggers(1, self.redPlayer)
self.assertEqual(2, self.board.person(self.redPlayer))
def testIsFinished(self):
self.board = Board([SimpleHut(Resource.wood,Resource.wood,Resource.clay),
SimpleHut(Resource.wood,Resource.wood,Resource.clay),
SimpleHut(Resource.wood,Resource.wood,Resource.clay),
SimpleHut(Resource.wood,Resource.wood,Resource.clay)])
self.assertFalse(self.board.isFinished())
self.board = Board([SimpleHut(Resource.wood,Resource.wood,Resource.clay),
SimpleHut(Resource.wood,Resource.wood,Resource.clay),
SimpleHut(Resource.wood,Resource.wood,Resource.clay)])
self.assertTrue(self.board.isFinished())
def testReapResources(self):
hutForRed = SimpleHut(Resource.wood, Resource.wood, Resource.clay)
hutForBlue = SimpleHut(Resource.wood, Resource.clay, Resource.clay)
self.board = Board([hutForRed, hutForBlue, SimpleHut(Resource.wood,Resource.clay,Resource.stone), SimpleHut(Resource.clay,Resource.stone,Resource.gold)])
self.board.placeOnHut(hutForRed, self.redPlayer)
self.board.placeOnHut(hutForBlue, self.bluePlayer)
huts, cards = self.board.reapResources(self.players)
self.assertEqual(1, len(huts))
self.assertEqual([hutForRed], huts)
self.players.reverse()
huts, cards = self.board.reapResources(self.players)
self.assertEqual([hutForBlue], huts)
def testBuyingFirstCard(self):
firstCards = self.board.cardPile[:4]
nextCards = self.board.cardPile[4:8]
self.assertListEqual(firstCards, self.board.openCards())
# buy the cheapest card
self.board.removeCards([firstCards[0]])
self.assertListEqual(firstCards[1:] + nextCards[:1], self.board.openCards())
def testBuyingFirstAndThirdCard(self):
firstCards = self.board.cardPile[:4]
nextCards = self.board.cardPile[4:8]
# buy the cheapest card
self.board.removeCards([firstCards[0], firstCards[2]])
self.assertListEqual([firstCards[1], firstCards[3]] + nextCards[:2], self.board.openCards())
def testReapResourcesWithFarm(self):
self.board.placeOnFarm(self.redPlayer)
self.board.addClayDiggers(4, self.redPlayer)
huts = self.board.reapResources(self.players)
self.assertEqual(1, self.redPlayer.getFoodTrack())
