Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import numpy as np | |
| import pytest | |
| from camguard.ui import EDITOR_WIDTH, editor_value, on_analyze, use_video | |
| from camguard.video import H264Writer | |
| # Que falte un paso (sin video, sin zona) no es un error de la app: tiene que mostrarse como | |
| # aviso y dejar los resultados como estaban, en vez de lanzar gr.Error. | |
| def video(tmp_path): | |
| path = tmp_path / "video.mp4" | |
| with H264Writer(path, fps=5, width=64, height=48) as writer: | |
| writer.write(np.zeros((48, 64, 3), np.uint8)) | |
| return str(path) | |
| # Fuera de la página, gr.Warning se emite como UserWarning: así se verifica que el aviso aparece. | |
| def test_analyze_without_video_warns_and_skips_outputs(): | |
| with pytest.warns(UserWarning, match="Primero subí un video"): | |
| assert list(on_analyze(None, None)) == [(gr.skip(),) * 3] | |
| def test_analyze_without_zone_warns_and_skips_outputs(video): | |
| with pytest.warns(UserWarning, match="Pintá la zona"): | |
| assert list(on_analyze(video, {"layers": []})) == [(gr.skip(),) * 3] | |
| def test_unreadable_upload_warns_and_keeps_everything(tmp_path): | |
| path = tmp_path / "roto.mp4" | |
| path.write_bytes(b"no es un video") | |
| with pytest.warns(UserWarning): | |
| assert use_video(str(path)) == (gr.skip(),) * 4 | |
| def test_editor_image_is_shrunk_keeping_proportions(): | |
| # Un cuadro grande en el editor cerraba la pestaña en iPhone; la zona se vuelve a agrandar | |
| # al tamaño del video en zone_from_layers. | |
| big = np.zeros((1080, 1920, 3), np.uint8) | |
| assert editor_value(big)["background"].shape == (540, EDITOR_WIDTH, 3) | |
| small = np.zeros((48, 64, 3), np.uint8) | |
| assert editor_value(small)["background"] is small | |