mirror of
https://github.com/Horizontal42/Sort_YM.git
synced 2026-11-10 15:42:02 +03:00
a8dd8d3bc8
Живой каталог client.genres() показал, что настоящий корень для рока - "allrock" (title "Рок"), а не "rock", как было угадано изначально. Из-за этого весь рок (559 из ~1900 треков, почти треть библиотеки) молча падал в "Разное". Таблица переписана по полному списку из 34 реальных корней (cache/genre_catalog.json), а не по предположениям об именах слагов.
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
import pytest
|
||
|
||
from sort_ym import genres
|
||
|
||
|
||
@pytest.fixture
|
||
def catalog():
|
||
# Форма реального дерева client.genres(): "rock"/"rusrock" - подгенры, реальный
|
||
# корень называется "allrock", а не "rock" (проверено на живом каталоге).
|
||
return {
|
||
"allrock": {"title": "Рок", "root_id": "allrock"},
|
||
"rock": {"title": "Иностранный рок", "root_id": "allrock"},
|
||
"rusrock": {"title": "Русский рок", "root_id": "allrock"},
|
||
"alternative": {"title": "Альтернатива", "root_id": "alternative"},
|
||
"rap": {"title": "Рэп и хип-хоп", "root_id": "rap"},
|
||
"trap": {"title": "Трэп", "root_id": "rap"},
|
||
}
|
||
|
||
|
||
@pytest.mark.parametrize(
|
||
"genre_raw,expected",
|
||
[
|
||
("allrock", "rock"),
|
||
("rock", "rock"),
|
||
("rusrock", "rock"),
|
||
("alternative", "indie-alt"),
|
||
("rap", "rap"),
|
||
("trap", "rap"),
|
||
(None, "other"),
|
||
("", "other"),
|
||
("totally-unknown-slug", "other"),
|
||
],
|
||
)
|
||
def test_bucket_for(catalog, genre_raw, expected):
|
||
assert genres.bucket_for(genre_raw, catalog) == expected
|
||
|
||
|
||
def test_flatten_catalog_walks_sub_genres():
|
||
class FakeGenre:
|
||
def __init__(self, id_, title, sub_genres=None):
|
||
self.id = id_
|
||
self.title = title
|
||
self.sub_genres = sub_genres or []
|
||
|
||
root = FakeGenre("rock", "Рок", [FakeGenre("rusrock", "Русский рок")])
|
||
|
||
flat = genres.flatten_catalog([root])
|
||
|
||
assert flat["rock"] == {"title": "Рок", "root_id": "rock"}
|
||
assert flat["rusrock"] == {"title": "Русский рок", "root_id": "rock"}
|