Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 29 additions & 0 deletions test/utils/test_video.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
import cv2
import numpy as np

from supervision.utils.video import process_video


def _create_test_video(path, num_frames, width=20, height=10):
fourcc = cv2.VideoWriter_fourcc(*"mp4v")
out = cv2.VideoWriter(path, fourcc, 1.0, (width, height))

for _ in range(num_frames):
frame = np.zeros((height, width, 3), dtype=np.uint8)
out.write(frame)

out.release()


def test_process_video_max_frames_exceeds_total_frames(tmp_path):
source_path = tmp_path / "source.mp4"
target_path = tmp_path / "target.mp4"

_create_test_video(str(source_path), num_frames=5)

process_video(
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Any assertion or other validation, then it runs?

source_path=str(source_path),
target_path=str(target_path),
callback=lambda frame, _: frame,
max_frames=10,
)
Comment on lines +24 to +29
Copy link

Copilot AI Jan 5, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This test lacks assertions to verify the expected behavior. The test should validate that:

  1. The target video file was created successfully
  2. The video was processed without raising an exception
  3. The target video contains the expected number of frames (5, not 10)

Consider adding assertions such as checking that the target file exists and verifying the frame count of the output video.

Copilot uses AI. Check for mistakes.
Loading