Update TestDAG2PAG.py

Signed-off-by: ZhengLi <[email protected]>

Author: zhengli0060

tests/TestDAG2PAG.py

@@ -87,12 +87,30 @@ def test_case_selection(self):
         pag = dag2pag(dag, islatent=[], isselection=[nodes[4]])
         print(pag)
 
+    def test_case_orient_rules(self):
+        nodes = []
+        X = {}
+        L = {}
+        for i in range(7):
+            node_name = f"X{i + 1}"
+            if i + 1 == 2:
+                node_name = f"L{i + 1}"
+            node = GraphNode(node_name)
+            nodes.append(node)
+            if i + 1 == 2:
+                L[2] = node
+            else:
+                X[i + 1] = node
+        dag = Dag(nodes)
+        dag.add_directed_edge(L[2], X[4])
+        dag.add_directed_edge(L[2], X[5])   
+        dag.add_directed_edge(L[2], X[6])
 
-
-
-if __name__ == "__main__":
-    test_model = TestDAG2PAG()
-    test_model.test_case1()
-    test_model.test_case2()
-    test_model.test_case3()
-    test_model.test_case_selection()
+        dag.add_directed_edge(X[5], X[7])
+        dag.add_directed_edge(X[1], X[4])
+        dag.add_directed_edge(X[1], X[7])
+        dag.add_directed_edge(X[3], X[7])   
+        pag = dag2pag(dag, [L[2]])
+        print(pag)
+        graphviz_pag = GraphUtils.to_pgv(pag)
+        graphviz_pag.draw("pag.png", prog='dot', format='png')