added file extension check to backendTest

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/backendTest.py b/tests/backendTest.py
index 470444c..31b3a99 100644
--- a/tests/backendTest.py
+++ b/tests/backendTest.py
@@ -68,6 +68,13 @@ f = os.path.join(cd,"static/uploads")
 #t = os.path.join(f,"receptor",target)
 #r = os.path.join(f,"ligands",ligand)
 #c = os.path.join(f,"configs",config)
+
+if ".pdbqt" not in receptor_name:
+    receptor_name+=".pdbqt"
+
+if ".pdbqt" not in ligand_name:
+    ligand_name+=".pdbqt"
+
 print(f)
 import tempfile
 from shutil import copy