def test_ofg_nested(self): m = models.OfgNested() f = th.function(m.inputs, m.outputs) pdf = PyDotFormatter() graph = pdf(f) assert len(graph.get_nodes()) == 7 assert len(graph.get_subgraph_list()) == 1 ofg1 = graph.get_subgraph_list()[0] assert len(ofg1.get_nodes()) == 6 assert len(ofg1.get_subgraph_list()) == 1 ofg2 = ofg1.get_subgraph_list()[0] assert len(ofg2.get_nodes()) == 4
def test_ofg_nested(self): m = models.OfgNested() f = th.function(m.inputs, m.outputs) self.check(f)