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