tests: unit: Fix test_command.py after new command addition

pypa · Jun 11, 2021 · dce8972 · dce8972
1 parent f6ce9a0
commit dce8972
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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)