diff --git a/tests/unit/test_commands.py b/tests/unit/test_commands.py index f34f7e5387b..cb144c5f6da 100644 --- a/tests/unit/test_commands.py +++ b/tests/unit/test_commands.py @@ -11,7 +11,8 @@ # These are the expected names of the commands whose classes inherit from # IndexGroupCommand. -EXPECTED_INDEX_GROUP_COMMANDS = ['download', 'install', 'list', 'wheel'] +EXPECTED_INDEX_GROUP_COMMANDS = [ + 'download', 'index', 'install', 'list', 'wheel'] def check_commands(pred, expected): @@ -49,7 +50,9 @@ def test_session_commands(): def is_session_command(command): return isinstance(command, SessionCommandMixin) - expected = ['download', 'install', 'list', 'search', 'uninstall', 'wheel'] + expected = [ + 'download', 'index', 'install', 'list', 'search', 'uninstall', 'wheel' + ] check_commands(is_session_command, expected)