def test_action(self): p = MockPlayer(1,'a',1000) p.set_action(["CALL:10","RAISE:20","FOLD:0"]) ans = ["CALL:10","RAISE:20","FOLD:0"] for i in range(3): eq_(ans[i], p.action(None))