Update __init__.py

Added new argument test_server_port to start the test_server