Separate test for to dot

Author: jmafoster1

causal_testing/specification/causal_dag.py

@@ -525,7 +525,10 @@ def to_dot(self) -> str:
         """Return a string of the DOT representation of the causal DAG.
         :return DOT string of the DAG.
         """
-        return str(nx.nx_pydot.to_pydot(self.graph))
+        dotstring = "digraph G {\n"
+        dotstring += "".join([f"{a} -> {b};\n" for a, b in self.graph.edges])
+        dotstring += "}"
+        return dotstring
 
     def __str__(self):
         return f"Nodes: {self.graph.nodes}\nEdges: {self.graph.edges}"

tests/json_front_tests/test_json_class.py

@@ -116,7 +116,6 @@ def test_generate_coefficient_tests_from_json(self):
                 }
             ]
         }
-        print(self.json_class.causal_specification.causal_dag.to_dot())
         self.json_class.test_plan = example_test
         effects = {"NoEffect": NoEffect()}
         estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}

tests/specification_tests/test_causal_dag.py

@@ -73,7 +73,7 @@ class TestCausalDAG(unittest.TestCase):
     def setUp(self) -> None:
         temp_dir_path = create_temp_dir_if_non_existent()
         self.dag_dot_path = os.path.join(temp_dir_path, "dag.dot")
-        dag_dot = """digraph G { A -> B; B -> C; D -> A; D -> C}"""
+        dag_dot = """digraph G { A -> B; B -> C; D -> A; D -> C;}"""
         f = open(self.dag_dot_path, "w")
         f.write(dag_dot)
         f.close()
@@ -99,6 +99,10 @@ def test_empty_casual_dag(self):
         causal_dag = CausalDAG()
         assert list(causal_dag.graph.nodes) == [] and list(causal_dag.graph.edges) == []
 
+    def test_to_dot(self):
+        causal_dag = CausalDAG(self.dag_dot_path)
+        self.assertEqual(causal_dag.to_dot(), """digraph G {\nA -> B;\nB -> C;\nD -> A;\nD -> C;\n}""")
+
     def tearDown(self) -> None:
         remove_temp_dir_if_existent()