Skip to content

Commit 67d1147

Browse files
rettiglzain-sohail
authored andcommitted
add config for loader tests
1 parent 285d878 commit 67d1147

File tree

5 files changed

+29
-5
lines changed

5 files changed

+29
-5
lines changed

sed/loader/generic/loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ class GenericLoader(BaseLoader):
2121
meta_handler (MetaHandler, optional): MetaHandler object. Defaults to None.
2222
"""
2323

24-
__name__ = "dask"
24+
__name__ = "generic"
2525

2626
supported_file_types = ["parquet", "csv", "json"]
2727

tests/loader/flash/test_config.yaml renamed to tests/data/loader/flash/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ core:
33

44
dataframe:
55
paths:
6-
data_raw_dir: raw/hdf/express-0/fl1user3
7-
data_parquet_dir: parquet
6+
data_raw_dir: ../data/loader/flash/
7+
data_parquet_dir: ../data/loader/flash/parquet
88
ubid_offset: 5
99
daq: fl1user3
1010
channels:

tests/data/loader/generic/config.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
test:

tests/data/loader/mpes/config.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
test:

tests/loader/test_loaders.py

Lines changed: 24 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import pytest
88
from _pytest.mark.structures import ParameterSet
99

10+
from sed.config import parse_config
1011
from sed.loader.base.loader import BaseLoader
1112
from sed.loader.loader_interface import get_loader
1213
from sed.loader.loader_interface import get_names_of_all_loaders
@@ -29,7 +30,17 @@ def get_loader_name_from_loader_object(loader: BaseLoader) -> str:
2930
str: extracted name.
3031
"""
3132
for loader_name in get_names_of_all_loaders():
32-
gotten_loader = get_loader(loader_name)
33+
gotten_loader = get_loader(
34+
loader_name,
35+
config=parse_config(
36+
os.path.join(
37+
test_data_dir,
38+
"loader",
39+
loader_name,
40+
"config.yaml",
41+
),
42+
),
43+
)
3344
if loader.__name__ is gotten_loader.__name__:
3445
return loader_name
3546
return ""
@@ -40,7 +51,18 @@ def get_all_loaders() -> List[ParameterSet]:
4051
loaders = []
4152

4253
for loader in [
43-
get_loader(x) for x in get_names_of_all_loaders() if x != "flash"
54+
get_loader(
55+
loader_name,
56+
config=parse_config(
57+
os.path.join(
58+
test_data_dir,
59+
"loader",
60+
loader_name,
61+
"config.yaml",
62+
),
63+
),
64+
)
65+
for loader_name in get_names_of_all_loaders()
4466
]:
4567
loaders.append(pytest.param(loader))
4668

0 commit comments

Comments
 (0)