github-personal/moma
1
0
Fork 0
mirror of https://github.com/shadoll/moma.git synced 2026-08-28 11:33:25 +00:00
Code Issues Packages Projects Releases 4 Wiki Activity

feat: Add HDR extraction from filename and update tests

Browse Source
This commit is contained in:
sha
2025-12-26 13:51:12 +00:00
parent 91df347727
commit 3ecffc0d3c
4 changed files with 31 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
renamer/test/filenames.txt
+2 -1
View File
@@ -208,4 +208,5 @@ The long title (2008) SD 720p UKR.avi
The long title (2008) UHD 1440p ENG.mp4
The long title (2008) UHD 1440 ENG.mp4
The long title (2008) 8K 4320p ENG.mp4
Troll 2 (2025) WEB-DL 2160p HDR Ukr Nor [Hurtom].mkv
Moana 2 (2024) MA WEB-DL 2160p SDR Ukr Eng [Hurtom].mkv
renamer/test/test_filename_extractor.py
+14 -1
View File
@@ -66,4 +66,17 @@ def test_extract_frame_class(filename):
assert isinstance(frame_class, str)
# Should be one of the valid frame classes or 'Unclassified'
valid_classes = set(FRAME_CLASSES.keys()) | {'Unclassified'}
assert frame_class in valid_classes
assert frame_class in valid_classes
@pytest.mark.parametrize("filename", load_test_filenames())
def test_extract_hdr(filename):
"""Test HDR extraction from filename"""
file_path = Path(filename)
extractor = FilenameExtractor(file_path)
hdr = extractor.extract_hdr()
# Print filename and extracted HDR clearly
print(f"\nFilename: \033[1;36m{filename}\033[0m")
print(f"Extracted HDR: \033[1;32m{hdr}\033[0m")
# HDR should be 'HDR' or None
assert hdr is None or hdr == 'HDR'