Update command_tests.py