def setUp(self): super(GroupIteratorTestBase, self).setUp() self.red = GroupMember('colors', 'red', lambda tgt: 'red' in tgt.name) self.green = GroupMember('colors', 'green', lambda tgt: 'green' in tgt.name) self.blue = GroupMember('colors', 'blue', lambda tgt: 'blue' in tgt.name)
def test_from_goal_valid(self): def predicate(tgt): return tgt == 42 goal = Goal('fred', action=lambda: None, group=Group('heathers', predicate)) self.assertEqual(GroupMember('heathers', 'fred', predicate), GroupMember.from_goal(goal))
def test_from_goal_invalid(self): with pytest.raises(ValueError): GroupMember.from_goal(Goal('fred', action=lambda: None))