Simplify `test_suite` class