def testCountAfterReaping(self): rs = Quarry() rs.addPerson(1, self.redPlayer) self.assertEqual(1, rs.count(self.redPlayer)) rs.reapResources(self.redPlayer) self.assertEqual(0, rs.count(self.redPlayer))