revises and renames file for testing scrape

tests/test_data/__init__.py

@@ -0,0 +1,34 @@
+import os
+import shutil
+
+from contextlib import contextmanager
+
+_TESTS_DIR = os.path.dirname(
+    os.path.dirname(os.path.realpath(__file__))
+)
+_TSV_PATH = f"{_TESTS_DIR}/tsv"
+_PHONES_PATH = f"{_TESTS_DIR}/phones"
+
+def write_files_dummy_phones_files(key: str, dialect: str):
+    with open(
+        f"{_PHONES_PATH}/{key}_{dialect}phonetic.phones",
+        "w",
+    ) as f1:
+        f1.write("a")
+    with open(
+        f"{_PHONES_PATH}/{key}_{dialect}phonemic.phones",
+        "w",
+    ) as f2:
+        f2.write("a")
+
+
+@contextmanager
+def handle_dummy_files(phones: bool, key: str, dialect: str):
+    os.mkdir(_TSV_PATH)
+    os.mkdir(_PHONES_PATH)
+    print("IN CONTENXT MAANGER")
+    if phones:
+        write_files_dummy_phones_files(key, dialect)
+    yield _TSV_PATH
+    shutil.rmtree(_TSV_PATH)
+    shutil.rmtree(_PHONES_PATH)

tests/test_data/test_scrape_file_naming.py → tests/test_data/test_scrape.py

@@ -5,8 +5,7 @@
 
 from data.src.scrape import _build_scraping_config
 
-
-PARENT_DIR = os.path.dirname(os.getcwd())
+from . import handle_dummy_files
 
 # "mul" should be a future-proof iso639 code to test with.
 # "mul" is resolved to "Multiple Languages" by iso639 package,
@@ -33,46 +32,27 @@
             {"key": "mul"},
             "test_",
             False,
-            [
-                "mul_test_phonetic.tsv",
-                "mul_test_phonemic.tsv",
-            ],
+            ["mul_test_phonetic.tsv", "mul_test_phonemic.tsv",],
         ),
         # Standard
         (
             {"key": "mul"},
             "",
             False,
-            [
-                "mul_phonetic.tsv",
-                "mul_phonemic.tsv",
-            ],
+            ["mul_phonetic.tsv", "mul_phonemic.tsv",],
         ),
     ],
 )
-def test_something(config_settings, dialect_suffix, phones, expected_file_name):
-    tsv_path = f"{PARENT_DIR}/tsv"
-    phones_path = f"{PARENT_DIR}/phones"
-    os.mkdir(tsv_path)
-    os.mkdir(phones_path)
-
-    if phones:
-        with open(
-            f"{phones_path}/{config_settings['key']}_{dialect_suffix}phonetic.phones",
-            "w",
-        ) as f1:
-            f1.write("a")
-        with open(
-            f"{phones_path}/{config_settings['key']}_{dialect_suffix}phonemic.phones",
-            "w",
-        ) as f2:
-            f2.write("a")
-    _build_scraping_config(
-        config_settings=config_settings, dialect_suffix=dialect_suffix
-    )
-    contents = os.listdir(tsv_path)
-    shutil.rmtree(tsv_path)
-    shutil.rmtree(phones_path)
+def test_file_creation(
+    config_settings, dialect_suffix, phones, expected_file_name
+):
+    with handle_dummy_files(
+        phones, config_settings["key"], dialect_suffix
+    ) as dummy_tsv_path:
+        _build_scraping_config(
+            config_settings=config_settings, dialect_suffix=dialect_suffix
+        )
+        tsv_contents = os.listdir(dummy_tsv_path)
 
-    for produced_file in contents:
+    for produced_file in tsv_contents:
         assert produced_file in expected_file_name