Updated test.py for integration tests.