def start_new_game(self) -> Game:
nodes = {}
for city in self.cities:
node = Node(city, self.infections[city])
connection = self.connections[city]
for link in connection:
if link in nodes:
target = nodes[link]
node.add_connection(target)
nodes[city] = node
return Game(nodes, self.diseases, [*self.infection_rate],
self.starting_city)
def init_nodes(self):
nodes = [Node(i) for i in range(54)]
for idx, node in enumerate(nodes):
boardPositions = nodeToHex[idx]
for boardIdx in boardPositions:
self.board[boardIdx].add_node(node)
for conn in connectionIdxToNodeIdx:
nodes[conn[0]].add_connection(nodes[conn[1]])
nodes[conn[1]].add_connection(nodes[conn[0]])
Board.add_harbors_to_nodes(nodes)
return nodes
def testNodeInitializedWithCertainParametersHasCorrectValues(self):
n = Node(5, 8, 132, 89, [3, 8, 9])
self.assertEqual(n.getName(), 5)
self.assertEqual(n.getSize(), 8)
self.assertEqual(n.getX(), 132)
self.assertEqual(n.getY(), 89)
self.assertSequenceEqual(n.getEdges(), [3, 8, 9])
def testNodeInitializedWithNoParametersHasCorrectValues(self):
unit = Node()
self.assertEqual(unit.getName(), 0)
self.assertEqual(unit.getSize(), 0)
self.assertEqual(unit.getX(), 0)
self.assertEqual(unit.getY(), 0)
self.assertEqual(unit.getEdges(), [])
def testNodeGetEdgesReturnsCorrectEdgeList(self):
unit = Node(5, 25, 125, 255, [9, 4, 2])
self.assertSequenceEqual(unit.getEdges(), [9, 4, 2])
def testNodeGetYReturnsCorrectYPosition(self):
unit = Node(5, 25, 125, 255, [9, 4, 2])
self.assertEquals(unit.getY(), 255)
def testNodeGetSizeReturnsCorrectSize(self):
unit = Node(5, 25, 125, 255, [9, 4, 2])
self.assertEquals(unit.getSize(), 25)
