def test_get_neighbors_of(self):
"""Находятся все соседи узлов из списка"""
xgml = processing.Xgml(TEST, IOC.config.mgen)
xgml.parse(str(IOC.config.mgen.xgml[TEST]))
grid = model.Grid(TEST, xgml.nodes, xgml.edges, IOC.config.mgen)
expected = tests.utils.get_nodes_keys([
grid.node(18),
grid.node(19),
grid.node(1),
grid.node(0),
grid.node(21),
grid.node(5),
grid.node(24),
grid.node(7),
grid.node(6),
grid.node(39),
grid.node(8),
grid.node(29),
grid.node(12),
grid.node(33),
grid.node(15),
grid.node(37),
grid.node(14),
grid.node(41),
grid.node(13)
])
# act
result = tests.utils.get_nodes_keys(
grid.get_neighbors_of(grid.border_nodes))
# assert
self.assertCountEqual(result, expected)
def _test_grid_capturing_stalingrad(self):
"""Выполняется захват в графе Сталинграда"""
xgml = processing.Xgml(STALIN, IOC.config.mgen)
xgml.parse(str(IOC.config.mgen.xgml[STALIN]))
grid = model.Grid(STALIN, xgml.nodes, xgml.edges, IOC.config.mgen)
path = pathlib.Path(r'./tmp/{}_{}.xgml'.format(STALIN, 0))
xgml.save_file(str(path), grid.nodes, grid.edges)
# Act
self.fail()
def _test_grid_capturing_moscow(self):
"""Выполняется захват в графе Москвы"""
xgml = processing.Xgml(MOSCOW, IOC.config.mgen)
xgml.parse(str(IOC.config.mgen.xgml[MOSCOW]))
grid = model.Grid(MOSCOW, xgml.nodes, xgml.edges, IOC.config.mgen)
path = pathlib.Path(r'./tmp/{}_{}.xgml'.format(MOSCOW, 0))
xgml.save_file(str(path), grid.nodes, grid.edges)
# Act
self.fail()
def test_border_stalin(self):
"""Упорядочиваются вершины линии фронта в сталинградском графе"""
# Arrange
xgml = processing.Xgml(STALIN, IOC.config.mgen)
xgml.parse(str(IOC.config.mgen.xgml[STALIN]))
grid = model.Grid(STALIN, xgml.nodes, xgml.edges, IOC.config.mgen)
expected = (209, 94, 93, 96, 101, 100, 99, 137, 139, 138, 157, 186,
163, 164, 165, 184, 183, 182, 194, 193, 177)
# Act
border = grid.border
# Assert
self.assertCountEqual(tuple(int(x.key) for x in border), expected)
self.assertSequenceEqual(tuple(int(x.key) for x in border), expected)
def test_border_test(self):
"""Упорядочиваются вершины линии фронта в тестовом графе"""
# Arrange
xgml = processing.Xgml(TEST, IOC.config.mgen)
xgml.parse(str(IOC.config.mgen.xgml[TEST]))
grid = model.Grid(TEST, xgml.nodes, xgml.edges, IOC.config.mgen)
# Act
frontline = grid.border_nodes
border = grid.border
# Assert
self.assertEqual(len(frontline), 9)
self.assertEqual(len(border), 9)
self.assertEqual(border[0].text, 'L1')
def test_confrontation_area_stalingrad_east(self):
"""Создаётся многоугольник восточной прифронтовой полосы"""
xgml = processing.Xgml(STALIN, MGEN)
xgml.parse(str(MGEN.xgml[STALIN]))
expected_keys = (192, 57, 196, 197, 185, 188, 187, 48, 110, 25, 102,
19, 95, 3, 191, 209, 94, 93, 96, 101, 100, 99, 137,
139, 138, 157, 186, 163, 164, 165, 184, 183, 182, 194,
193, 177)
builder = processing.BoundaryBuilder(self.north, self.east, self.south,
self.west)
grid = model.Grid(STALIN, xgml.nodes, xgml.edges, 1)
path = pathlib.Path(r'./tmp/{}_{}.xgml'.format(STALIN, 0))
xgml.save_file(str(path), grid.nodes, grid.edges)
# Act
result = builder.confrontation_east(grid)
# Assert
self.assertSequenceEqual(tuple(int(x.key) for x in result),
expected_keys)
def test_grid_capturing_test(self):
"""Выполняется захват в тестовом графе"""
xgml = processing.Xgml(TEST, IOC.config.mgen)
xgml.parse(str(IOC.config.mgen.xgml[TEST]))
grid = model.Grid(TEST, xgml.nodes, xgml.edges, IOC.config.mgen)
path = pathlib.Path(r'./tmp/{}_{}.xgml'.format(TEST, 0))
xgml.save_file(str(path), grid.nodes, grid.edges)
# Act
for i in range(1, self.iterations):
nodes = list(grid.border_nodes)
if len(nodes) == 0:
break
random.shuffle(nodes)
for node in nodes.pop().neighbors:
if node.country == 201:
node.capture(101)
break
path = pathlib.Path(r'./tmp/{}_{}.xgml'.format(TEST, i))
xgml.save_file(str(path), grid.nodes, grid.edges)