def test_consume_add_arg(self): fsm = FSM() fsm.newState("A") fsm.newState("B") fsm.setStartState("A") def _test(*args): self.assertEqual(args[-1], "test") fsm.addTransition("A", "B", "input1", _test, "foo") fsm.consume("input1", "test")