def testCount(self):
rs = River()
rs.addPerson(1, self.redPlayer)
self.assertEqual(1, rs.count(self.redPlayer))
with self.assertRaisesRegex(PlacementError, "Player %s already added person to the River" % self.redPlayer.getColor()):
rs.addPerson(1, self.redPlayer)
self.assertEqual(1, rs.count(self.redPlayer))
def testCountTwoPlayers(self):
rs = River()
rs.addPerson(1, self.redPlayer)
self.assertEqual(1, rs.count(self.redPlayer))
self.assertEqual(6, rs.freeSlots())
rs.addPerson(3, self.bluePlayer)
self.assertEqual(3, rs.count(self.bluePlayer))
self.assertEqual(3, rs.freeSlots())
