This commit is contained in:
2026-02-04 08:52:10 -05:00
parent c33c1d2f36
commit 6d1df4dede
2 changed files with 8 additions and 5 deletions

View File

@@ -9,7 +9,7 @@ from typing import Optional, Dict
import typer import typer
from rich.console import Console from rich.console import Console
from rich.progress import Progress, SpinnerColumn, TextColumn from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
from rich.table import Table from rich.table import Table
import time import time
@@ -186,8 +186,10 @@ def generate(
with Progress( with Progress(
SpinnerColumn(), SpinnerColumn(),
TextColumn("[progress.description]{task.description}"), TextColumn("[progress.description]{task.description}"),
TimeElapsedColumn(),
console=console console=console
) as progress: ) as progress:
progress.add_task("Elapsed", total=None)
for shot in storyboard_data.shots: for shot in storyboard_data.shots:
task_id = progress.add_task(f"Shot {shot.id}", total=None) task_id = progress.add_task(f"Shot {shot.id}", total=None)

View File

@@ -113,6 +113,7 @@ class TestGenerateCommand:
]) ])
assert result.exit_code == 0 assert result.exit_code == 0
assert "Dry run complete" in result.output assert "Dry run complete" in result.output
assert "Elapsed:" in result.output
def test_generate_nonexistent_storyboard(self): def test_generate_nonexistent_storyboard(self):
"""Test generate with non-existent storyboard.""" """Test generate with non-existent storyboard."""