mirror of
https://github.com/alexta69/metube.git
synced 2026-09-21 21:45:04 +00:00
Your GitHub Name
120 lines
3.3 KiB
Python
120 lines
3.3 KiB
Python
"""Tests for conservative audio metadata enrichment."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from music_metadata import MusicMetadataPreProcessor
|
|
|
|
|
|
def _preprocess(source_entry, info):
|
|
processor = MusicMetadataPreProcessor(source_entry=source_entry)
|
|
_, result = processor.run(info)
|
|
return result
|
|
|
|
|
|
def test_album_uses_existing_order_and_total_when_track_number_is_missing():
|
|
result = _preprocess(
|
|
{
|
|
'playlist_index': '03',
|
|
'playlist_count': 12,
|
|
'playlist_title': 'Example Album',
|
|
},
|
|
{'title': 'Track', 'album': 'Example Album'},
|
|
)
|
|
|
|
assert result['track_number'] == '3/12'
|
|
assert result['album'] == 'Example Album'
|
|
|
|
|
|
def test_official_track_number_wins_over_album_order():
|
|
result = _preprocess(
|
|
{'playlist_index': 3, 'playlist_count': 12},
|
|
{'track_number': 7, 'album': 'Official Album'},
|
|
)
|
|
|
|
assert result['track_number'] == '7/12'
|
|
assert result['album'] == 'Official Album'
|
|
|
|
|
|
def test_inline_official_track_total_is_preserved():
|
|
result = _preprocess(
|
|
{'playlist_count': 12},
|
|
{'track_number': '4/10', 'album': 'Official Album'},
|
|
)
|
|
|
|
assert result['track_number'] == '4/10'
|
|
|
|
|
|
def test_source_track_number_and_total_are_retained_from_flat_extraction():
|
|
result = _preprocess(
|
|
{'track_number': 2, 'track_count': 9, 'playlist_index': 4},
|
|
{'title': 'Track'},
|
|
)
|
|
|
|
assert result['track_number'] == '2/9'
|
|
|
|
|
|
def test_album_title_falls_back_to_source_playlist_title():
|
|
result = _preprocess(
|
|
{'playlist_title': 'Example Album'},
|
|
{'track_number': 4},
|
|
)
|
|
|
|
assert result['album'] == 'Example Album'
|
|
assert result['track_number'] == 4
|
|
|
|
|
|
def test_playlist_without_extractor_album_signals_is_not_changed():
|
|
result = _preprocess(
|
|
{
|
|
'playlist_index': 3,
|
|
'playlist_count': 12,
|
|
'playlist_title': 'Example Playlist',
|
|
},
|
|
{'title': 'Track'},
|
|
)
|
|
|
|
assert 'album' not in result
|
|
assert 'track_number' not in result
|
|
|
|
|
|
def test_regular_video_artwork_is_not_changed():
|
|
thumbnails = [
|
|
{'url': 'square.jpg', 'width': 500, 'height': 500},
|
|
{'url': 'landscape.jpg', 'width': 1280, 'height': 720},
|
|
]
|
|
result = _preprocess({}, {'title': 'Regular Video', 'thumbnails': thumbnails.copy()})
|
|
|
|
assert result['thumbnails'] == thumbnails
|
|
assert 'thumbnail' not in result
|
|
|
|
|
|
def test_music_audio_prefers_largest_existing_square_thumbnail():
|
|
result = _preprocess(
|
|
{},
|
|
{
|
|
'track': 'Track',
|
|
'thumbnails': [
|
|
{'url': 'small-square.jpg', 'width': 200, 'height': 200},
|
|
{'url': 'large-square.jpg', 'width': 1000, 'height': 1000},
|
|
{'url': 'landscape.jpg', 'width': 1280, 'height': 720},
|
|
],
|
|
},
|
|
)
|
|
|
|
assert result['thumbnails'][-1]['url'] == 'large-square.jpg'
|
|
assert result['thumbnail'] == 'large-square.jpg'
|
|
|
|
|
|
def test_landscape_only_music_artwork_keeps_existing_order():
|
|
thumbnails = [
|
|
{'url': 'small.jpg', 'width': 640, 'height': 360},
|
|
{'url': 'large.jpg', 'width': 1280, 'height': 720},
|
|
]
|
|
result = _preprocess(
|
|
{},
|
|
{'track': 'Track', 'thumbnails': thumbnails.copy()},
|
|
)
|
|
|
|
assert result['thumbnails'] == thumbnails
|
|
assert 'thumbnail' not in result
|