Rewrite integration test suite in Python