def test_get_pos(self):
id = 10
pos = (1, 2, 0)
node = Node(id)
node.setPos(pos)
x, y, z = node.getPos()
self.assertTrue(x == 1)
self.assertTrue(y == 2)
def testPos(self): n1 = Node(0) n2 = Node(0) self.assertEqual(n1.getPos(), n2.getPos()) n1.setPos((0, 0, 1)) self.assertNotEqual(n1.getPos(), n2.getPos()) n2.setPos((0, 0, 1)) self.assertEqual(n1.getPos(), n2.getPos()) n1.incPos(5) self.assertEqual(n1.getPos(), (0, 0, 50))
def add_node(self, node_id: int, pos: tuple = None):
if self.__Nodes.get(node_id) is None:
node = Node(node_id)
node.setPos(pos)
self.__Nodes.update({node_id: node})
self.__Edges.update({node_id: {}})
self.__allInEdges.update({node_id: {}})
self.__nodeSize += 1
self.__mc += 1
if node_id + 1 > self.__maxKey:
self.__maxKey = node_id + 1
return True
return False
