def test_as_single_sample(): ds = get_ds(tag='highmass') sample = as_single_sample(ds) assert sample['mS'] == '275' assert sample['mH'] == '1000' assert sample['lifetime'] == '5' assert sample['campaign'] == 'mc16a,mc16d,mc16e' assert sample['tags'] == 'signal,highmass' assert isinstance(sample['data'], xaod_table)
def test_as_samples(): ds = get_ds(mH=125, mS=35, lifetime=5, campaign='mc16e') s = as_samples(ds) assert len(s) == 1 assert "mH" in s[0] assert "mS" in s[0] assert "lifetime" in s[0] assert 'campaign' in s[0] assert 'tags' in s[0] assert s[0]["mH"] == 125 assert s[0]["mS"] == 35 assert s[0]["lifetime"] == 5 assert s[0]["campaign"] == 'mc16e' assert isinstance(s[0]['data'], xaod_table)
def test_default_get_returns_no_non_signal(): ds = as_samples(get_ds()) for d in ds: assert 'signal' in d['tags']
def test_ds_no_args(): ds = get_ds() assert len(ds) > 10
def test_as_jz_workers(): ds = get_ds(tag='jz3') sample = as_single_sample(ds) assert isinstance(sample['data'], xaod_table) assert sample['data'].event_source[0]._ds._max_workers == 200
def test_ds_get_tag(): ds = get_ds(tag="highmass") assert len(ds) == 3
def test_ds_get_one(): ds = get_ds(mH=125, mS=35, lifetime=5, campaign='mc16e') assert len(ds) == 1