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.testNoise()""")