refactor code

OpenPecha · Oct 21, 2024 · ddd271b · ddd271b
1 parent ad951a2
commit ddd271b
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/tests/prepare/test_prepare.py b/tests/prepare/test_prepare.py
@@ -1,3 +1,4 @@
+import tempfile
 from pathlib import Path
 
 from bo_rag_prep_tool.prepare import prepare
@@ -8,12 +9,14 @@ def test_prepare():
     data = Path(__file__).parent / "data"
     opf_path = data / "I46409446.opf"
 
-    output_path = Path(__file__).parent / "output"
-    json_output_path = prepare(opf_path, output_path)
+    with tempfile.TemporaryDirectory() as tmpdirname:
+        output_path = Path(tmpdirname)
 
-    expected_output = read_json(data / "expected_output.json")
+        json_output_path = prepare(opf_path, output_path)
 
-    json_output = read_json(json_output_path)
-    assert len(json_output) == len(expected_output)
-    for json_data in json_output:
-        assert json_data in expected_output
+        expected_output = read_json(data / "expected_output.json")
+
+        json_output = read_json(json_output_path)
+        assert len(json_output) == len(expected_output)
+        for json_data in json_output:
+            assert json_data in expected_output