change: allow passing datadir to tests.common.load_data_for_testfile

ssato · May 20, 2024 · f0f4033 · f0f4033
1 parent a7ef11c
commit f0f4033
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/common/tdc.py b/tests/common/tdc.py
@@ -60,14 +60,17 @@ def load_data_for_testfile(
     testfile: str,
     values: tuple[tuple[str, typing.Optional[dict], ...], ...] = VALUES,
     load_idata: bool = False,
+    datadir: typing.Optional[pathlib.Path] = None,
     **opts
 ) -> list[tuple[pathlib.Path, dict[str, typing.Any], ...]]:
     """Collct test data for test file, ``testfile``.
 
     :param testfile: a str represents test file path
     :param opts: keyword options for `get_test_resdir`
     """
-    datadir = get_test_resdir(testfile, **opts)
+    if datadir is None:
+        datadir = get_test_resdir(testfile, **opts)
+
     if load_idata:
         return [
             (ipath, idata, *[aux.get(k, v) for k, v in values])