handle non-json file for get_data function

neoave · Feb 4, 2024 · af31717 · af31717
1 parent ab412cc
commit af31717
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/utils.py b/tests/unit/utils.py
@@ -11,7 +11,10 @@ def test_dir_path():
 def get_data(name, data_dir="data"):
     path = os.path.join(test_dir_path(), data_dir, name)
     with open(os.path.expanduser(path), "r", encoding="utf-8") as file_data:
-        data = json.load(file_data)
+        if name.endswith("json"):
+            data = json.load(file_data)
+        else:
+            data = file_data.read()
     return data