Update test_suite.py