Function test_save_load_from_video

tests/test_stats_manager.py:137–168

Test generating and saving some frame metrics from TEST_VIDEO_FILE to a file on disk, and loading the file back to ensure the loaded frame metrics agree with those that were saved.

(test_video_file, tmp_path: Path)

def test_save_load_from_video(test_video_file, tmp_path: Path):
    """Test generating and saving some frame metrics from TEST_VIDEO_FILE to a file on disk, and
    loading the file back to ensure the loaded frame metrics agree with those that were saved.
    """
    video = VideoStreamCv2(test_video_file)
    stats_manager = StatsManager()
    scene_manager = SceneManager(stats_manager)

    scene_manager.add_detector(ContentDetector())

    video_fps = video.frame_rate
    duration = FrameTimecode("00:00:05", video_fps)

    scene_manager.auto_downscale = True
    scene_manager.detect_scenes(video, duration=duration)

    path = tmp_path.joinpath("stats.csv")
    stats_manager.save_to_csv(csv_file=path)

    metrics = stats_manager.metric_keys

    stats_manager_new = StatsManager()

    stats_manager_new.load_from_csv(path)

    # Compare the first 5 frames. Frame 0 won't have any metrics for this detector.
    for frame in range(1, 5 + 1):
        assert stats_manager.metrics_exist(frame, metrics)
        orig_metrics = stats_manager.get_metrics(frame, metrics)
        new_metrics = stats_manager_new.get_metrics(frame, metrics)
        for i, metric_val in enumerate(orig_metrics):
            assert metric_val == pytest.approx(new_metrics[i])


def test_load_corrupt_stats(tmp_path: Path):