Update cli_tests.py