def test_gen_sort(self):
aliases = list(gen.fs_content())
tmplt = gen.filter(aliases, {"analyzer": "fakeTemplate"})
sorted = list(gen.sort(tmplt))
s_name = map(lambda x: x.name, sorted)
s_sample = map(lambda x: x.sample, sorted)
s_is_data = map(lambda x: x.is_data, sorted)
self.assertTrue(s_sample.index("ttgamma") < s_sample.index("zjets"))
self.assertTrue(s_name.index("sihihEB") < s_name.index("sihihEE"))
self.assertTrue(s_is_data.index(False) < s_is_data.index(True))
def test_gen_sort(self):
aliases = list(gen.fs_content())
tmplt = gen.filter(aliases, {"analyzer": "fakeTemplate"})
sorted = list(gen.sort(tmplt))
s_name = map(lambda x: x.name, sorted)
s_sample = map(lambda x: x.sample, sorted)
s_is_data = map(lambda x: x.is_data, sorted)
self.assertTrue(s_sample.index("ttgamma") < s_sample.index("zjets"))
self.assertTrue(s_name.index("sihihEB") < s_name.index("sihihEE"))
self.assertTrue(s_is_data.index(False) < s_is_data.index(True))
def test_gen_group(self):
from_fs = gen.fs_content()
filtered = gen.filter(from_fs, {"name": "histo"})
sorted = gen.sort(filtered)
grouped = gen.group(sorted)
group_list = []
for group in grouped:
group_list.append(list(group))
# length: 3 samples: 3 histos per group
self.assertEqual(len(group_list), 26)
for g in group_list:
self.assertEqual(len(g), 3)
self.assertEqual(g[0].analyzer, g[1].analyzer)
self.assertEqual(g[0].analyzer, g[2].analyzer)
def test_gen_group(self):
from_fs = gen.fs_content()
filtered = gen.filter(from_fs, {"name": "histo"})
sorted = gen.sort(filtered)
grouped = gen.group(sorted)
group_list = []
for group in grouped:
group_list.append(list(group))
# length: 3 samples: 3 histos per group
self.assertEqual(len(group_list), 26)
for g in group_list:
self.assertEqual(len(g), 3)
self.assertEqual(g[0].analyzer, g[1].analyzer)
self.assertEqual(g[0].analyzer, g[2].analyzer)
