from unit import Unit from constants import * from graph import Graph warpling = Unit() warpling.type = WARPLING_TYPE warpling.name = "Warpling" warpling.cost = 0 warpling.moves = Graph() warpling.moves.add_new_node(0, 0) warpling.moves.add_new_node(0, 1) warpling.moves.connect_adjacent_nodes() knight = Unit() knight.type = KNIGHT_TYPE knight.name = "Knight" knight.cost = 3 knight.moves = Graph() knight.moves.add_new_node(0, 0) knight.moves.add_new_node(1, 2) knight.moves.add_new_node(-1, 2) knight.moves.add_new_node(2, 1) knight.moves.add_new_node(2, -1) knight.moves.add_new_node(1, -2) knight.moves.add_new_node(-1, -2) knight.moves.add_new_node(-2, -1) knight.moves.add_new_node(-2, 1) knight.moves.connect_all_to(knight.moves.find_node_by_position(0, 0)) bishop = Unit() bishop.type = BISHOP_TYPE