Merge pull request #27 from lfoppiano/test-tests

.github/workflows/ci-build.yml

@@ -34,9 +34,9 @@ jobs:
           flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
           # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
           flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
-#    - name: Test with pytest
-#      run: |
-#        pytest
+      - name: Test with pytest
+        run: |
+          pytest
 
   docker-build:
     needs: [build]

tests/resources/__init__.py

@@ -0,0 +1,3 @@
+import os
+
+TEST_DATA_PATH = os.path.dirname(__file__)

tests/test_grobid_processors.py

@@ -1,9 +1,12 @@
+import os
+
 from bs4 import BeautifulSoup
 from document_qa.grobid_processors import get_xml_nodes_body, get_xml_nodes_figures, get_xml_nodes_header
+from tests.resources import TEST_DATA_PATH
 
 
 def test_get_xml_nodes_body_paragraphs():
-    with open("resources/2312.07559.paragraphs.tei.xml", 'r') as fo:
+    with open(os.path.join(TEST_DATA_PATH, "2312.07559.paragraphs.tei.xml"), 'r') as fo:
         soup = BeautifulSoup(fo, 'xml')
 
     nodes = get_xml_nodes_body(soup, use_paragraphs=True)
@@ -12,7 +15,7 @@ def test_get_xml_nodes_body_paragraphs():
 
 
 def test_get_xml_nodes_body_sentences():
-    with open("resources/2312.07559.sentences.tei.xml", 'r') as fo:
+    with open(os.path.join(TEST_DATA_PATH, "2312.07559.sentences.tei.xml"), 'r') as fo:
         soup = BeautifulSoup(fo, 'xml')
 
     children = get_xml_nodes_body(soup, use_paragraphs=False)
@@ -21,7 +24,7 @@ def test_get_xml_nodes_body_sentences():
 
 
 def test_get_xml_nodes_figures():
-    with open("resources/2312.07559.paragraphs.tei.xml", 'r') as fo:
+    with open(os.path.join(TEST_DATA_PATH, "2312.07559.paragraphs.tei.xml"), 'r') as fo:
         soup = BeautifulSoup(fo, 'xml')
 
     children = get_xml_nodes_figures(soup)
@@ -30,17 +33,18 @@ def test_get_xml_nodes_figures():
 
 
 def test_get_xml_nodes_header_paragraphs():
-    with open("resources/2312.07559.paragraphs.tei.xml", 'r') as fo:
+    with open(os.path.join(TEST_DATA_PATH, "2312.07559.paragraphs.tei.xml"), 'r') as fo:
         soup = BeautifulSoup(fo, 'xml')
 
     children = get_xml_nodes_header(soup)
 
-    assert len(children) == 8
+    assert sum([len(child) for k, child in children.items()]) == 8
+
 
 def test_get_xml_nodes_header_sentences():
-    with open("resources/2312.07559.sentences.tei.xml", 'r') as fo:
+    with open(os.path.join(TEST_DATA_PATH, "2312.07559.sentences.tei.xml"), 'r') as fo:
         soup = BeautifulSoup(fo, 'xml')
 
     children = get_xml_nodes_header(soup, use_paragraphs=False)
 
-    assert len(children) == 15
+    assert sum([len(child) for k, child in children.items()]) == 15