import sys
import cProfile

from tests.testEngine import TestTransitions
from tests.testStatebag import TestStateBag


arg=sys.argv[1]

if arg=="engine":
	t=TestTransitions()
	cProfile.run(r"""t.testReal()""")
elif arg=="statebag":
	t=TestStateBag()
	cProfile.run(r"""t.testReal()""")