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