Create socket_test.py