Fixed test issues

shaunhutch · shaunhutch · commit f8437ee4d491 · 2023-06-27T13:18:04.000-07:00
diff --git a/tests/data_review_tool/test_article_review.py b/tests/data_review_tool/test_article_review.py
@@ -7,6 +7,9 @@
 # ensure that the parent directory is on the path for relative imports
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
+os.environ["ARTICLE_RELEVANCE_BATCH"] = "article-relevance-output.parquet"
+os.environ["ENTITY_EXTRACTION_BATCH"] = "entity-extraction-output.zip"
+
 from src.data_review_tool.app import *
 from src.data_review_tool.pages.article_review import *
 
@@ -22,20 +25,6 @@ def test_cell_clicked():
     assert cell_clicked(0) == dash.no_update
 
 
-def test_directory_structure():
-    "Test that the directory structure is as expected."
-    dir = "data/data-review-tool"
-    expected = ["processed", "raw"]
-    assert expected[0] in os.listdir(dir) and expected[1] in os.listdir(dir)
-
-
-def test_find_start_end_char():
-    "Test that the find_start_end_char function returns the correct values."
-    text = "This is a test"
-    entity = "test"
-    assert find_start_end_char(text, entity) == (10, 14)
-
-
 def test_update_button():
     "Test that the update_button function returns correct children for Delete or Restore."
     assert update_button(True)[0].children == "Delete Entity"
diff --git a/tests/data_review_tool/test_home.py b/tests/data_review_tool/test_home.py
@@ -10,20 +10,15 @@
 
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
+os.environ["ARTICLE_RELEVANCE_BATCH"] = "article-relevance-output.parquet"
+os.environ["ENTITY_EXTRACTION_BATCH"] = "entity-extraction-output.zip"
+
 # Import the names of callback functions you want to test
 from src.data_review_tool.app import *
 from src.data_review_tool.pages.home import *
 from src.data_review_tool.pages.config import *
 
 
-def test_directory_structure():
-    "test that the data directory structure is correct"
-    dir = "data/data-review-tool"
-    expected = ["processed", "raw"]
-
-    assert expected[0] in os.listdir(dir) and expected[1] in os.listdir(dir)
-
-
 def test_current_article_clicked():
     "test that clicking 'Review' button redirects to the correct article"
     active_cell_current = AttributeDict({"row": 0, "column_id": "Review"})
