mirror of
https://github.com/shadoll/moma.git
synced 2026-08-28 03:27:34 +00:00
sha
b50b9bc165
- Added core caching functionality with `Cache` class supporting in-memory and file-based caching. - Introduced `CacheManager` for high-level cache operations and statistics. - Created various cache key generation strategies: `FilepathMethodStrategy`, `APIRequestStrategy`, `SimpleKeyStrategy`, and `CustomStrategy`. - Developed decorators for easy method caching: `cached`, `cached_method`, `cached_api`, and `cached_property`. - Implemented type definitions for cache entries and statistics. - Added comprehensive tests for cache operations, strategies, and decorators to ensure functionality and backward compatibility.
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""Test script for MediaInfo frame class detection by resolution"""
|
|
|
|
import json
|
|
import pytest
|
|
from unittest.mock import MagicMock
|
|
import sys
|
|
import os
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
|
|
|
|
from renamer.extractors.mediainfo_extractor import MediaInfoExtractor
|
|
from pathlib import Path
|
|
|
|
# Load test cases from JSON file using context manager
|
|
test_cases_file = Path(__file__).parent / 'test_mediainfo_frame_class.json'
|
|
with open(test_cases_file, 'r') as f:
|
|
test_cases = json.load(f)
|
|
|
|
@pytest.mark.parametrize("test_case", test_cases, ids=[tc['testname'] for tc in test_cases])
|
|
def test_frame_class_detection(test_case):
|
|
"""Test frame class detection for various resolutions"""
|
|
|
|
testname = test_case['testname']
|
|
width, height = test_case['resolution']
|
|
interlaced = test_case['interlaced']
|
|
expected = test_case['expected_frame_class']
|
|
|
|
# Create a mock MediaInfoExtractor
|
|
extractor = MagicMock(spec=MediaInfoExtractor)
|
|
from pathlib import Path
|
|
extractor.file_path = Path(f"test_{testname}") # Set a unique file_path for caching
|
|
|
|
# Mock the video_tracks
|
|
mock_track = MagicMock()
|
|
mock_track.height = height
|
|
mock_track.width = width
|
|
mock_track.interlaced = 'Yes' if interlaced else 'No'
|
|
|
|
extractor.video_tracks = [mock_track]
|
|
|
|
# Test the method
|
|
actual = MediaInfoExtractor.extract_frame_class(extractor)
|
|
|
|
assert actual == expected, f"{testname}: expected {expected}, got {actual}" |