Added missing import in test_cli.py.

test_cli.py is just a boilerplate unit test for now.