def test_dict(self): event = Event(0) event.the_dict = dict([(x, x**2) for x in range(10)]) cfg_ana = cfg.Analyzer(Selector, output='filtered', input_objects='the_dict', filter_func=lambda x: x == 9) cfg_comp = cfg.Component('test', files=[]) filter = Selector(cfg_ana, cfg_comp, self.outdir) filter.process(event) self.assertDictEqual(event.filtered, {3: 9})
def test_list(self): event = Event(0) event.the_list = range(10) cfg_ana = cfg.Analyzer(Selector, output='filtered', input_objects='the_list', filter_func=lambda x: x % 2 == 0) cfg_comp = cfg.Component('test', files=[]) filter = Selector(cfg_ana, cfg_comp, self.outdir) filter.process(event) self.assertItemsEqual(event.filtered, [0, 2, 4, 6, 8])
def test_dict(self): event = Event(0) event.the_dict = dict( [ (x, x**2) for x in range(10) ] ) cfg_ana = cfg.Analyzer( Filter, output = 'filtered', input_objects = 'the_dict', filter_func = lambda x : x == 9 ) cfg_comp = cfg.Component( 'test', files = [] ) filter = Filter(cfg_ana, cfg_comp, self.outdir) filter.process(event) self.assertDictEqual(event.filtered, {3:9})
def test_list(self): event = Event(0) event.the_list = range(10) cfg_ana = cfg.Analyzer( Filter, output = 'filtered', input_objects = 'the_list', filter_func = lambda x : x%2 == 0 ) cfg_comp = cfg.Component( 'test', files = [] ) filter = Filter(cfg_ana, cfg_comp, self.outdir) filter.process(event) self.assertItemsEqual(event.filtered, [0,2,4,6,8])