def testPrint(self): n1 = Node(1) n2 = Node(2) n3 = Node(3) n2.setStart(n1.getId(), 12) n2.setEnd(n3.getId(), 23) self.assertEqual(n2.__repr__(), "2: |edges out| 1 |edges in| 1")
def testGetId(self): n1 = Node(0) n2 = Node(0) self.assertEqual(n1.getId(), n2.getId()) self.assertEqual(n1.getId(), 0) n2 = Node(1) self.assertNotEqual(n1.getId(), n2.getId())
def testJson(self): n1 = Node(1) n2 = Node(2) n3 = Node(3) n2.setStart(n1.getId(), 12) n2.setEnd(n3.getId(), 23) self.assertEqual(n2.toJson(), {'id': 2})
def hasNext(self, node: Node) -> bool: edges = self.graph.all_out_edges_of_node(node.getId()) s = [] for i in edges: if (not self.graph.get_node(i).getStatus()): s.append(i) if (len(s) != 0): return True return False
def getNext(self, node: Node) -> Node: edges = self.graph.all_out_edges_of_node(node.getId()) for i in edges: if (not self.graph.get_node(i).getStatus()): return self.graph.get_node(i) pass