| """Tests for IngestAgent.""" | |
| import pytest | |
| import numpy as np | |
| from formscout.agents.ingest import IngestAgent | |
| from formscout.types import IngestResult | |
| def _make_test_video(path, fps=30, n_frames=30, w=640, h=480): | |
| """Create a minimal test video using OpenCV.""" | |
| import cv2 | |
| out = cv2.VideoWriter(str(path), cv2.VideoWriter_fourcc(*"mp4v"), fps, (w, h)) | |
| for _ in range(n_frames): | |
| out.write(np.zeros((h, w, 3), dtype=np.uint8)) | |
| out.release() | |
| class TestIngestAgent: | |
| def test_returns_typed_result(self, tmp_path): | |
| p = tmp_path / "test.mp4" | |
| _make_test_video(p) | |
| agent = IngestAgent() | |
| result = agent.run(str(p)) | |
| assert isinstance(result, IngestResult) | |
| assert result.fps == pytest.approx(30.0, abs=2.0) | |
| assert len(result.frames) > 0 | |
| assert result.width == 640 | |
| assert result.height == 480 | |
| assert result.confidence == 1.0 | |
| def test_rejects_missing_file(self): | |
| agent = IngestAgent() | |
| result = agent.run("/nonexistent/path.mp4") | |
| assert result.confidence == 0.0 | |
| assert "not found" in result.notes.lower() | |
| def test_result_is_frozen(self, tmp_path): | |
| p = tmp_path / "test.mp4" | |
| _make_test_video(p, n_frames=10, w=64, h=64) | |
| agent = IngestAgent() | |
| result = agent.run(str(p)) | |
| with pytest.raises(Exception): | |
| result.fps = 999.0 | |
| def test_caps_frames(self, tmp_path): | |
| """Verify frame cap works on long videos.""" | |
| p = tmp_path / "long.mp4" | |
| _make_test_video(p, n_frames=600) | |
| agent = IngestAgent() | |
| result = agent.run(str(p)) | |
| assert len(result.frames) <= 300 | |