def test_raise(self): obj = LambdaStr(lambda_str = 'ev: aaa', name = 'test_true') event = MockEvent() obj.begin(event) event = MockEvent() self.assertRaises(ValueError, obj, event) obj.end()
def test_standard(self): obj = LambdaStr(lambda_str = 'ev : True', name = 'test_true') event = MockEvent() obj.begin(event) self.assertTrue(callable(obj.func)) event = MockEvent() self.assertTrue(obj(event)) event = MockEvent() self.assertTrue(obj.event(event)) obj.end() self.assertIsNone(obj.func)
dict(Not=dict( Any=( 'ev : ev.z[0] == 0', 'ev : ev.w[0] >= 300', 'alias5', 'alias6', ), )), )), dict( AllClass=All, AnyClass=Any, NotClass=Not, LambdaStrClass=LambdaStr, aliasDict=alias_dict, n=5242, ), Any(selections=[ LambdaStr('ev : ev.x[0] == 0'), All(selections=[ LambdaStr('ev : ev.x[0] >= 1'), LambdaStr('ev : ev.y[0] >= 100'), LambdaStr(name='alias4', lambda_str='ev : ev.var1[0] >= 10'), ]), Not(selection=Any(selections=[ LambdaStr('ev : ev.z[0] == 0'), LambdaStr('ev : ev.w[0] >= 300'), LambdaStr(name='alias5', lambda_str='ev : ev.var4[0] == 5242'), LambdaStr(name='alias6', lambda_str='ev : 11 <= ev.var5[0] < 20'), ]), ), ]), id='example' ),
from alphatwirl.selection.factories.factory import FactoryDispatcher from alphatwirl.selection.modules.LambdaStr import LambdaStr from alphatwirl.selection.modules import All, Any, Not ##__________________________________________________________________|| # path_cfg, expanded, obj params = [ pytest.param('ev : ev.nJets[0] >= 2', dict( components=(), factory='LambdaStrFactory', lambda_str='ev : ev.nJets[0] >= 2', ), LambdaStr( name='ev : ev.nJets[0] >= 2', lambda_str='ev : ev.nJets[0] >= 2', ), id='string:lambda_str'), pytest.param('ev : ev.nJets[0] >= {n}', dict( components=(), factory='LambdaStrFactory', lambda_str='ev : ev.nJets[0] >= {n}', ), LambdaStr( name='ev : ev.nJets[0] >= 5242', lambda_str='ev : ev.nJets[0] >= 5242', ), id='string:lambda_str-not-formatted'), pytest.param(dict(All=()), { 'factory': 'AllFactory',
'alias6': ('ev : {low} <= ev.var5[0] < {high}', dict(low=11, high=20)) } ##__________________________________________________________________|| # path_cfg, expanded, obj params = [ pytest.param( 'alias1', dict( factory='LambdaStrFactory', components=(), lambda_str='ev : ev.var1[0] >= 10', name='alias1' ), LambdaStr( name='alias1', lambda_str='ev : ev.var1[0] >= 10' ), id='alias1' ), pytest.param( ('alias1', dict(name='name1')), dict( factory='LambdaStrFactory', components=(), lambda_str='ev : ev.var1[0] >= 10', name='name1' ), LambdaStr(name='name1', lambda_str='ev : ev.var1[0] >= 10'), id='alias1:with-name' ), pytest.param(