Complete VIDEO-001 through VIDEO-004 software video studies

tests/test_video003_transport.py

@@ -0,0 +1,62 @@
+"""Проверка временной модели VIDEO-003 на настоящих малых H.264 кадрах."""
+
+from __future__ import annotations
+
+import math
+
+import av
+import numpy as np
+import pytest
+
+from protocol.video_stream_transport import FrameType, VideoFrame
+from tools.video002_simulation import simulate
+from tools.video_compare import new_h264_encoder
+from tools.video003_design import estimate
+
+
+def _frames() -> list[VideoFrame]:
+    encoder, _ = new_h264_encoder(64, 48, 5, 160_000)
+    output = []
+    for index in range(10):
+        pixels = np.zeros((48, 64, 3), dtype=np.uint8)
+        pixels[:, :, 1] = index * 17
+        pixels[8:20, 3 * index:3 * index + 12, 0] = 240
+        raw = av.VideoFrame.from_ndarray(pixels, format="rgb24").reformat(format="yuv420p")
+        raw.pts = index
+        packets = encoder.encode(raw)
+        assert len(packets) == 1
+        packet = packets[0]
+        output.append(VideoFrame(33, index,
+                                 FrameType.KEYFRAME if packet.is_keyframe
+                                 else FrameType.DELTA_FRAME,
+                                 200 * index, bytes(packet)))
+    assert not encoder.encode(None)
+    return output
+
+
+def test_explicit_fec_airtime_controls_video_age_and_p95() -> None:
+    frames = _frames()
+    row = estimate("QPSK", 500_000, "2/3", 1012, 0)
+    link = ("QPSK_500_R23", "QPSK", 500_000, 1012, 0)
+    calculated = simulate(frames, 5, link, ("random", 0),
+                          airtime_override_ms=row.packet_airtime_ms)
+    slower = simulate(frames, 5, link, ("random", 0),
+                      airtime_override_ms=row.packet_airtime_ms * 2)
+    assert calculated["displayed_frames"] == 10
+    assert slower["displayed_frames"] == 10
+    assert calculated["p95_end_to_end_age_ms"] <= slower["p95_end_to_end_age_ms"]
+    assert calculated["p95_end_to_end_age_ms"] <= calculated["max_end_to_end_age_ms"]
+    assert calculated["no_old_frame_replay"] and slower["no_old_frame_replay"]
+    with pytest.raises(ValueError):
+        simulate(frames, 5, link, ("random", 0), airtime_override_ms=math.nan)
+
+
+def test_design_headroom_uses_actual_video001_peaks() -> None:
+    from tools.video003_design import measured_video_profiles
+
+    profiles = measured_video_profiles()
+    assert profiles["NORMAL"]["measured_peak_1s_bps"] == pytest.approx(375_776, abs=200)
+    assert profiles["HIGH_QUALITY"]["measured_peak_1s_bps"] == pytest.approx(455_600, abs=200)
+    for profile in profiles.values():
+        assert profile["required_20pct_bps"] == pytest.approx(
+            profile["measured_peak_1s_bps"] * 1.2)