test: fix tests

dhdaines · Nov 22, 2023 · af644f2 · af644f2
1 parent 950515a
commit af644f2
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/test/data/model.gz b/test/data/model.gz
diff --git a/test/test_analyse.py b/test/test_analyse.py
@@ -6,7 +6,7 @@
 from alexi.format import format_xml
 
 DATADIR = Path(__file__).parent / "data"
-TRAINDIR = Path(__file__).parent.parent / "data" / "train"
+TRAINDIR = Path(__file__).parent.parent / "data"
 
 IOBTEST = [
     "<Titre>Titre incomplet</Titre>",

diff --git a/test/test_format.py b/test/test_format.py
@@ -6,7 +6,7 @@
 from alexi.format import format_html, format_text
 
 DATADIR = Path(__file__).parent / "data"
-TRAINDIR = Path(__file__).parent.parent / "data" / "train"
+TRAINDIR = Path(__file__).parent.parent / "data"
 
 
 def test_format_html():

diff --git a/test/test_segment.py b/test/test_segment.py
@@ -17,7 +17,7 @@ def test_segment():
                 del word["segment"]
                 writer.writerow(word)
             testfh.seek(0, 0)
-            seg = Segmenteur()
+            seg = Segmenteur(DATADIR / "model.gz")
             reader = csv.DictReader(testfh)
             words = list(seg(reader))
     assert len(words) > 0