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

feat: Add movie database identifier extraction and update related tests

Browse Source
This commit is contained in:
sha
2025-12-26 14:34:58 +00:00
parent 38e3f0e553
commit c5ab985440
6 changed files with 84 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
renamer/test/test_filename_extractor.py
+18 -1
View File
@@ -79,4 +79,21 @@ def test_extract_hdr(filename):
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'
assert hdr is None or hdr == 'HDR'
@pytest.mark.parametrize("filename", load_test_filenames())
def test_extract_movie_db(filename):
"""Test movie database identifier extraction from filename"""
file_path = Path(filename)
extractor = FilenameExtractor(file_path)
movie_db = extractor.extract_movie_db()
# Print filename and extracted movie DB clearly
print(f"\nFilename: \033[1;36m{filename}\033[0m")
print(f"Extracted movie DB: \033[1;32m{movie_db}\033[0m")
# Movie DB should be tuple (str, str) or None
if movie_db:
assert isinstance(movie_db, tuple) and len(movie_db) == 2
assert isinstance(movie_db[0], str) and isinstance(movie_db[1], str)
else:
assert movie_db is None