Update websocket_test.py

Taking out the assertion for error number as this may not be the same on all platforms and may cause the test to fail. Checking for type of exception should suffice.