fix(tests): update CLI tests for current command structure

- Update command list: remove non-existent (src, export, test, grade, logo)
- Add actual commands: dev, milestone (singular), olympics
- Fix 'milestones' → 'milestone' throughout all CLI tests
- Update expected command files for orphan detection

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: profvjreddi

tinytorch/tests/cli/test_cli_execution.py

@@ -69,9 +69,8 @@ def test_tito_version(self):
         assert "Tiny" in result.stdout or "CLI" in result.stdout
 
     @pytest.mark.parametrize("command", [
-        'setup', 'system', 'module', 'src', 'package', 'nbgrader',
-        'milestones', 'benchmark', 'community', 'export', 'test',
-        'grade', 'logo'
+        'setup', 'system', 'module', 'dev', 'package', 'nbgrader',
+        'milestone', 'benchmark', 'community', 'olympics'
     ])
     def test_command_help_works(self, command):
         """Test that each command's help can be displayed."""
@@ -99,7 +98,7 @@ def test_command_help_works(self, command):
         ('module', 'status'),
         ('module', 'list'),
         ('community', 'join'),
-        ('milestones', 'status'),
+        ('milestone', 'status'),
     ])
     def test_subcommand_help_works(self, command, subcommand):
         """Test that subcommands can show help."""
@@ -135,7 +134,7 @@ def test_student_facing_commands_discoverable(self):
         )
 
         # Key student commands should be visible
-        student_commands = ['setup', 'module', 'milestones']
+        student_commands = ['setup', 'module', 'milestone']
 
         for cmd in student_commands:
             assert cmd in result.stdout, (
@@ -152,7 +151,7 @@ def test_developer_commands_documented(self):
         )
 
         # Developer commands should be in help
-        dev_commands = ['src', 'package', 'nbgrader']
+        dev_commands = ['dev', 'package', 'nbgrader']
 
         for cmd in dev_commands:
             assert cmd in result.stdout, (

tinytorch/tests/cli/test_cli_help_consistency.py

@@ -38,7 +38,7 @@ def get_command_help(self, *args):
 
     def test_all_command_helps_mention_tito(self):
         """Verify all help texts mention 'tito' command."""
-        commands = ['setup', 'system', 'module', 'checkpoint', 'milestones']
+        commands = ['setup', 'system', 'module', 'milestone']
 
         for cmd in commands:
             help_text = self.get_command_help(cmd)
@@ -68,7 +68,7 @@ def test_bare_tito_and_help_are_different(self):
         help_output = help_result.stdout
 
         # Should both mention key commands
-        for cmd in ['module', 'milestones', 'setup']:
+        for cmd in ['module', 'milestone', 'setup']:
             assert cmd in bare_output, f"'{cmd}' missing from bare tito"
             assert cmd in help_output, f"'{cmd}' missing from tito -h"
 
@@ -207,8 +207,8 @@ def test_all_registered_commands_in_welcome_or_help(self):
             f"Add them to welcome screen or help text"
         )
 
-    def test_milestones_properly_documented(self):
-        """Specifically test that milestones command is documented."""
+    def test_milestone_properly_documented(self):
+        """Specifically test that milestone command is documented."""
         # This addresses the user's concern about progress tracking
         welcome_result = subprocess.run(
             [sys.executable, '-m', 'tito.main'],
@@ -226,21 +226,21 @@ def test_milestones_properly_documented(self):
 
         combined = welcome_result.stdout + help_result.stdout
 
-        assert 'milestones' in combined.lower(), \
-            "milestones command should be documented"
+        assert 'milestone' in combined.lower(), \
+            "milestone command should be documented"
 
         # Check it has a description
-        milestones_help = subprocess.run(
-            [sys.executable, '-m', 'tito.main', 'milestones', '-h'],
+        milestone_help = subprocess.run(
+            [sys.executable, '-m', 'tito.main', 'milestone', '-h'],
             cwd=self.project_root,
             capture_output=True,
             text=True
         )
 
-        assert 'progress' in milestones_help.stdout.lower() or \
-               'track' in milestones_help.stdout.lower() or \
-               'milestone' in milestones_help.stdout.lower(), \
-            "milestones help should explain its purpose"
+        assert 'progress' in milestone_help.stdout.lower() or \
+               'track' in milestone_help.stdout.lower() or \
+               'milestone' in milestone_help.stdout.lower(), \
+            "milestone help should explain its purpose"
 
 
 if __name__ == '__main__':

tinytorch/tests/cli/test_cli_registry.py

@@ -134,20 +134,13 @@ def test_command_files_exist(self):
             'setup': 'setup.py',
             'system': 'system/__init__.py',  # Now in subfolder
             'module': 'module/__init__.py',   # Now in subfolder
-            'src': 'src.py',
+            'dev': 'dev/__init__.py',         # Now in subfolder
             'package': 'package/__init__.py', # Now in subfolder
             'nbgrader': 'nbgrader.py',
-            'milestones': 'milestone.py',
-            'leaderboard': 'leaderboard.py',
+            'milestone': 'milestone.py',      # Singular, not plural
             'olympics': 'olympics.py',
             'benchmark': 'benchmark.py',
             'community': 'community.py',
-            'export': 'export.py',
-            'test': 'test.py',
-            'book': 'book.py',
-            'grade': 'grade.py',
-            'demo': 'demo.py',
-            'logo': 'logo.py',
         }
 
         for cmd_name, expected_file in cmd_to_file.items():
@@ -168,14 +161,12 @@ def test_no_orphaned_command_files(self):
         # Files we expect to see (registered commands + internal helpers)
         expected_files = {
             # Registered top-level commands
-            'setup.py', 'src.py', 'nbgrader.py',
+            'setup.py', 'nbgrader.py',
             'milestone.py', 'benchmark.py',
-            'community.py', 'export.py', 'test.py',
-            'grade.py', 'logo.py',
-            # Known internal/subcommand files (not top-level)
+            'community.py', 'olympics.py',
+            # Known internal/helper files (not top-level commands)
             'login.py',  # Subcommand of community
-            'clean_workspace.py', 'version.py', 'check.py', 'view.py',
-            'protect.py', 'report.py'
+            'export_utils.py',  # Helper for export functionality
         }
 
         orphaned = []
@@ -209,8 +200,7 @@ def test_epilog_mentions_registered_commands(self):
             'module',
             'package',
             'nbgrader',
-            'milestones',
-            'leaderboard',
+            'milestone',
             'olympics'
         ]