Use console_shell in test/main.py instead of console