update tests

Author: yifanfeng97

pyproject.toml

@@ -6,7 +6,7 @@ authors = ["yifanfeng97 <[email protected]>"]
 license = "Apache-2.0"
 readme = "README.md"
 homepage = "https://deephypergraph.com"
-repository = "https://github.com/imoonlab/deephypergraph"
+repository = "https://github.com/iMoonLab/DeepHypergraph"
 
 [tool.poetry.dependencies]
 python = "^3.8"

tests/structure/test_bigraph.py

@@ -24,6 +24,8 @@ def test_save(g1):
     from pathlib import Path
 
     save_path = Path("./tmp/test_save.dhg")
+    if not save_path.parent.exists():
+        save_path.parent.mkdir()
     g1.save(save_path)
     g2 = load_structure(save_path)
 

tests/structure/test_digraph.py

@@ -26,6 +26,8 @@ def test_save(g1):
     from pathlib import Path
 
     save_path = Path("./tmp/test_save.dhg")
+    if not save_path.parent.exists():
+        save_path.parent.mkdir()
     g1.save(save_path)
     g2 = load_structure(save_path)
 

tests/visualization/test_feature.py

@@ -1,5 +1,6 @@
 import pytest
 import numpy as np
+from pathlib import Path
 
 import dhg.visualization as vis
 
@@ -10,14 +11,20 @@ def emb1():
 
 
 def test_draw_in_euclidean_space(emb1):
+    tmp_path = Path("tmp/")
+    if not tmp_path.exists():
+        tmp_path.mkdir()
     fig = vis.draw_in_euclidean_space(emb1)
-    fig.savefig("tmp/test_draw_in_euclidean_space.png")
+    fig.savefig(tmp_path / "test_draw_in_euclidean_space.png")
     fig = vis.draw_in_euclidean_space(emb1, dim=3)
-    fig.savefig("tmp/test_draw_in_euclidean_space_3d.png")
+    fig.savefig(tmp_path / "test_draw_in_euclidean_space_3d.png")
 
 
 def test_draw_poincare_ball(emb1):
+    tmp_path = Path("tmp/")
+    if not tmp_path.exists():
+        tmp_path.mkdir()
     fig = vis.draw_in_poincare_ball(emb1)
-    fig.savefig("tmp/test_draw_poincare_ball.png")
+    fig.savefig(tmp_path / "test_draw_poincare_ball.png")
     fig = vis.draw_in_poincare_ball(emb1, dim=3)
-    fig.savefig("tmp/test_draw_poincare_ball_3d.png")
+    fig.savefig(tmp_path / "test_draw_poincare_ball_3d.png")