Update test_message.py