fix: bug

HephaestusProject · Sep 8, 2020 · 609d798 · 609d798
1 parent 40c5005
commit 609d798
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
@@ -1,5 +1,5 @@
 import pytest  # noqa: E902
-from dataset import WMT14Dataset
+from src.dataset import WMT14Dataset
 
 
 @pytest.mark.parametrize("langpair", ["en-de"])

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -1,10 +1,12 @@
 import pytest  # noqa: F401
+from pathlib import Path
+
 from src.utils import read_lines
 
-test_filepath = "data/example.de"
+test_filepath = Path(__file__, "data/example.de").parent.absolute()
 
 
-@pytest.mark.parametrize("filepath", test_filepath)
+@pytest.mark.parametrize("filepath", [test_filepath])
 def test_read_lines(filepath):
     de = read_lines(filepath)
     assert isinstance(de, list)