class Game(object):
'''
class to represent the game loop
'''
def __init__(self):
self.board = Board()
self.players = []
def playerCount(self):
return len(self.players)
def addPlayer(self, player):
self.players.append(player)
def allPersonsPlaced(self):
return sum([player.personsLeft(self.board) for player in self.players]) == 0
def processRound(self):
for player in self.players:
player.newRound()
while not self.allPersonsPlaced():
for player in [p for p in self.players if p.personsLeft(self.board) > 0]:
# print("Player: %-6s to place persons" % (player.getOutputColor()))
if isinstance(player.strategy, Human):
print (self.board)
player.placePersons(self.board)
print("Player: %-6s to reap resources " % (player.getOutputColor()))
for idx, player in enumerate(self.players): # reap resources and buy building tiles
if isinstance(player.strategy, Human):
print (self.board)
activeFirst = self.players[idx:] + self.players[:idx]
huts, cards = self.board.reapResources(activeFirst)
boughtCards = player.buyCards(cards, activeFirst, self.board.cardPile)
self.board.removeCards(boughtCards)
boughtHuts = player.buyHuts(huts)
self.board.popHuts(boughtHuts)
if isinstance(player.strategy, Human):
print(player)
for player in self.players: # feed and adjust score
player.feed()
# cycle player sequence
self.players = self.players[1:] + self.players[:1]
def finished(self):
return self.board.isFinished()
def getPlayers(self):
return sorted(self.players[:], reverse = True)
def doPrintPlayers(self, heading, scoreFunc):
print(heading)
maximalColorLength = max(len(player.getColor().name) for player in self.players)
for player in self.getPlayers():
print("%s : %s (%s)" % (player.getOutputColor(maximalColorLength), scoreFunc(player) , player.getStrategy()))
def printPlayers(self):
self.doPrintPlayers("Players:", lambda p: "")
def printScores(self):
self.doPrintPlayers("Scores:", lambda p: "%3d" % p.getScore())
def printFinalScores(self):
self.doPrintPlayers("\n\nFinal scores:", lambda p: "%3d, second criteria: %2d" % (p.getScore(), p.secondScoreCriteria()))
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())
