Skip to content

Commit 537271d

Browse files
committed
Adding test to eos path URL resolution
1 parent e990741 commit 537271d

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

tests/test_dataset_resolver.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,3 +45,18 @@
4545
def test_find_dataset(input_ds, expected_type):
4646
dataset = ds_type_resolver(input_ds)
4747
assert isinstance(dataset, expected_type)
48+
49+
50+
@pytest.mark.parametrize(
51+
"eos_path, prefix",
52+
[
53+
("/eos/opendata/atlas/rucio/somefile.root", "root://eospublic.cern.ch/"),
54+
("/eos/opendata/cms/rucio/somefile.root", "root://eospublic.cern.ch/"),
55+
("/eos/atlas/atlascerngroupdisk/somefile.root", "root://eosatlas.cern.ch/"),
56+
("/eos/cms/store/somefile.root", "root://eoscms.cern.ch/"),
57+
],
58+
)
59+
def test_eos_url_parsing(eos_path, prefix):
60+
ds_out = ds_type_resolver(eos_path)
61+
assert isinstance(ds_out, dataset.FileList)
62+
assert ds_out.files[0] == prefix + eos_path

0 commit comments

Comments
 (0)