def test_attacked_locations(self): knight = Knight(Location(5, 'd'), Player.BLACK, self.board) results = {Location(7, 'c'), Location(6, 'b'), Location(4, 'b'), Location(3, 'c'), Location(3, 'e'), Location(4, 'f'), Location(6, 'f'), Location(7, 'e')} self.assertEquals(set(knight.attacked_locations()), results)