added tests for input files

Author: yucongalicechen

src/diffpy/labpdfproc/tests/conftest.py

@@ -15,4 +15,16 @@ def user_filesystem(tmp_path):
     existing_file = Path(tmp_path).resolve() / "existing_file.py"
     existing_file.touch()
 
+    input_directory = Path(tmp_path).resolve() / "input_directory"
+    input_directory.mkdir(parents=True, exist_ok=True)
+
+    with open(os.path.join(input_directory, "good_data.chi"), "w") as f:
+        f.write("1 2 \n 3 4 \n 5 6 \n 7 8")
+    with open(os.path.join(input_directory, "good_data.xy"), "w") as f:
+        f.write("1 2 \n 3 4 \n 5 6 \n 7 8")
+    with open(os.path.join(input_directory, "unreadable_file.txt"), "w") as f:
+        f.write("This is an unreadable file.")
+    with open(os.path.join(input_directory, "binary.pkl"), "wb") as f:
+        f.write(b"\x00\x01\x02\x03\x04")
+
     yield tmp_path

src/diffpy/labpdfproc/tests/test_tools.py

@@ -1,10 +1,12 @@
 import argparse
+import os
 import re
 from pathlib import Path
 
 import pytest
 
 from diffpy.labpdfproc.tools import known_sources, set_input_directory, set_output_directory, set_wavelength
+from diffpy.utils.parsers.loaddata import loadData
 
 params1 = [
     ([None], ["."]),
@@ -29,7 +31,6 @@ def test_set_output_directory(inputs, expected, user_filesystem):
 
 
 def test_set_output_directory_bad(user_filesystem):
-    # tmp_dir = user_filesystem
     actual_parser = argparse.ArgumentParser()
     actual_parser.add_argument("--output_directory")
     actual_args = actual_parser.parse_args(["--output_directory", "existing_file.py"])
@@ -60,6 +61,16 @@ def test_set_input_directory(inputs, expected, user_filesystem):
     assert actual_args.input_directory == expected_input_directory
 
 
+def test_input_files(user_filesystem):
+    os.chdir("input_directory")
+    xarray_chi, yarray_chi = loadData("good_data.chi", unpack=True)
+    xarray_xy, yarray_xy = loadData("good_data.xy", unpack=True)
+    with pytest.raises(ValueError):
+        xarray_txt, yarray_txt = loadData("unreadable_file.txt", unpack=True)
+    with pytest.raises(ValueError):
+        xarray_pkl, yarray_pkl = loadData("binary.pkl", unpack=True)
+
+
 params2 = [
     ([None, None], [0.71]),
     ([None, "Ag"], [0.59]),