[Python Trik Test] Changes in stress test function