class TestThing(TestCase):
def setUp(self):
self.thing = Thing(name='thing')
self.action1 = Action.create(name='action1')
self.action2 = Action.create(name='action2')
self.thing.add_agent_role(verb='action1')
def test_add_role(self):
self.thing.add_agent_role(verb='action2')
self.assertTrue(self.thing.is_valid_agent(self.action2))
def test_is_valid_for_role(self):
self.thing.add_role(Role.THEME, 'action2')
self.assertTrue(self.thing.is_valid_for_role(Role.THEME, self.action2))
