convert file paths to pathlib type

KosinskiLab · Dec 20, 2023 · 5d4eee6 · 5d4eee6
1 parent 08482dd
commit 5d4eee6
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/test_custom_db.py b/test/test_custom_db.py
@@ -1,5 +1,5 @@
 import pytest
-from alphapulldown.create_custom_template_db import create_db, parse_code
+from alphapulldown.create_custom_template_db import create_db
 import tempfile
 import os
 from alphafold.data import mmcif_parsing
@@ -63,13 +63,13 @@ def run_test(pdb_templates, chains):
 
 
 def test_from_pdb(capfd):
-    run_test(["./test/test_data/true_multimer/3L4Q.pdb"], ["C"])
+    run_test([Path("./test/test_data/true_multimer/3L4Q.pdb")], ["C"])
 
 def test_from_cif(capfd):
-    run_test(["./test/test_data/true_multimer/3L4Q.cif"], ["A"])
+    run_test([Path("./test/test_data/true_multimer/3L4Q.cif")], ["A"])
 
 def test_from_af_output_pdb(capfd):
-    run_test(["./test/test_data/true_multimer/cage_BC_AF.pdb"], ["B"])
+    run_test([Path("./test/test_data/true_multimer/cage_BC_AF.pdb")], ["B"])
 
 def test_from_minimal_pdb(capfd):
-    run_test(["./test/test_data/true_multimer/RANdom_name1_.7-1_0.pdb"], ["B"])
+    run_test([Path("./test/test_data/true_multimer/RANdom_name1_.7-1_0.pdb")], ["B"])