def test_attacked_locations(self):
queen = Queen(Location(5, 'd'), Player.BLACK, self.board)
results = {Location(4, 'c'), Location(3, 'b'), Location(2, 'a'), Location(4, 'e'), Location(3, 'f'),
Location(2, 'g'), Location(1, 'h'), Location(6, 'c'), Location(7, 'b'), Location(8, 'a'),
Location(6, 'e'), Location(7, 'f'), Location(8, 'g'), Location(4, 'd'), Location(3, 'd'),
Location(2, 'd'), Location(1, 'd'), Location(6, 'd'), Location(7, 'd'), Location(8, 'd'),
Location(5, 'c'), Location(5, 'b'), Location(5, 'a'), Location(5, 'e'), Location(5, 'f'),
Location(5, 'g'), Location(5, 'h')}
self.assertEquals(set(queen.attacked_locations()), results)
