Add the path to the source code to sys.path in tests

This is needed so that pytest can actually import dgmr to run the tests
Mews · Jun 11, 2024 · 568992d · 568992d
1 parent b018594
commit 568992d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/tests/test_losses.py b/tests/test_losses.py
@@ -1,4 +1,12 @@
 """ Test loss functions"""
+
+import os
+import sys
+
+sys.path.insert(
+    0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+)
+
 from dgmr.losses import SSIMLoss, MS_SSIMLoss, SSIMLossDynamic, tv_loss
 import torch
 

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -1,3 +1,10 @@
+import os
+import sys
+
+sys.path.insert(
+    0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+)
+
 import torch
 import torch.nn.functional as F
 from dgmr import (