def test_iterate_pieces_with_opponent_pieces(self): b = Board().place_piece(p1).place_piece(p2).place_piece(opponent) self.assertEqual(len(list(b.iterate_pieces(Owner.PLAYER))), 2) self.assertEqual(len(list(b.iterate_pieces(Owner.OPPONENT))), 1)
def test_iterate_pieces_with_opponent_pieces(self): b = Board().place_piece(p1).place_piece(p2).place_piece(opponent) self.assertEqual(len(list(b.iterate_pieces(Owner.PLAYER))), 2) self.assertEqual(len(list(b.iterate_pieces(Owner.OPPONENT))), 1)