mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 08:39:17 +01:00
autotag: split tests into test_match.py and test_hooks.py
This commit is contained in:
Šarūnas Nejus
parent
2789a9d342
commit
05bc57f1fb
2 changed files with 94 additions and 89 deletions
|
|
@ -27,99 +27,10 @@ from beets.autotag.hooks import (
|
|||
TrackMatch,
|
||||
correct_list_fields,
|
||||
)
|
||||
from beets.autotag.match import assign_items
|
||||
from beets.library import Item
|
||||
from beets.test.helper import ConfigMixin
|
||||
|
||||
|
||||
class TestAssignment(ConfigMixin):
|
||||
A = "one"
|
||||
B = "two"
|
||||
C = "three"
|
||||
|
||||
@pytest.fixture(scope="class", autouse=True)
|
||||
def _setup_config(self):
|
||||
self.config["match"]["track_length_grace"] = 10
|
||||
self.config["match"]["track_length_max"] = 30
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
# 'expected' is a tuple of expected (mapping, extra_items, extra_tracks)
|
||||
"item_titles, track_titles, expected",
|
||||
[
|
||||
# items ordering gets corrected
|
||||
([A, C, B], [A, B, C], ({A: A, B: B, C: C}, [], [])),
|
||||
# unmatched tracks are returned as 'extra_tracks'
|
||||
# the first track is unmatched
|
||||
([B, C], [A, B, C], ({B: B, C: C}, [], [A])),
|
||||
# the middle track is unmatched
|
||||
([A, C], [A, B, C], ({A: A, C: C}, [], [B])),
|
||||
# the last track is unmatched
|
||||
([A, B], [A, B, C], ({A: A, B: B}, [], [C])),
|
||||
# unmatched items are returned as 'extra_items'
|
||||
([A, C, B], [A, C], ({A: A, C: C}, [B], [])),
|
||||
],
|
||||
)
|
||||
def test_assign_tracks(self, item_titles, track_titles, expected):
|
||||
expected_mapping, expected_extra_items, expected_extra_tracks = expected
|
||||
|
||||
items = [Item(title=title) for title in item_titles]
|
||||
tracks = [TrackInfo(title=title) for title in track_titles]
|
||||
|
||||
item_info_pairs, extra_items, extra_tracks = assign_items(items, tracks)
|
||||
|
||||
assert (
|
||||
{i.title: t.title for i, t in item_info_pairs},
|
||||
[i.title for i in extra_items],
|
||||
[t.title for t in extra_tracks],
|
||||
) == (expected_mapping, expected_extra_items, expected_extra_tracks)
|
||||
|
||||
def test_order_works_when_track_names_are_entirely_wrong(self):
|
||||
# A real-world test case contributed by a user.
|
||||
def item(i, length):
|
||||
return Item(
|
||||
artist="ben harper",
|
||||
album="burn to shine",
|
||||
title=f"ben harper - Burn to Shine {i}",
|
||||
track=i,
|
||||
length=length,
|
||||
)
|
||||
|
||||
items = []
|
||||
items.append(item(1, 241.37243007106997))
|
||||
items.append(item(2, 342.27781704375036))
|
||||
items.append(item(3, 245.95070222338137))
|
||||
items.append(item(4, 472.87662515485437))
|
||||
items.append(item(5, 279.1759535763187))
|
||||
items.append(item(6, 270.33333768012))
|
||||
items.append(item(7, 247.83435613222923))
|
||||
items.append(item(8, 216.54504531525072))
|
||||
items.append(item(9, 225.72775379800484))
|
||||
items.append(item(10, 317.7643606963552))
|
||||
items.append(item(11, 243.57001238834192))
|
||||
items.append(item(12, 186.45916150485752))
|
||||
|
||||
def info(index, title, length):
|
||||
return TrackInfo(title=title, length=length, index=index)
|
||||
|
||||
trackinfo = []
|
||||
trackinfo.append(info(1, "Alone", 238.893))
|
||||
trackinfo.append(info(2, "The Woman in You", 341.44))
|
||||
trackinfo.append(info(3, "Less", 245.59999999999999))
|
||||
trackinfo.append(info(4, "Two Hands of a Prayer", 470.49299999999999))
|
||||
trackinfo.append(info(5, "Please Bleed", 277.86599999999999))
|
||||
trackinfo.append(info(6, "Suzie Blue", 269.30599999999998))
|
||||
trackinfo.append(info(7, "Steal My Kisses", 245.36000000000001))
|
||||
trackinfo.append(info(8, "Burn to Shine", 214.90600000000001))
|
||||
trackinfo.append(info(9, "Show Me a Little Shame", 224.0929999999999))
|
||||
trackinfo.append(info(10, "Forgiven", 317.19999999999999))
|
||||
trackinfo.append(info(11, "Beloved One", 243.733))
|
||||
trackinfo.append(info(12, "In the Lord's Arms", 186.13300000000001))
|
||||
|
||||
expected = list(zip(items, trackinfo)), [], []
|
||||
|
||||
assert assign_items(items, trackinfo) == expected
|
||||
|
||||
|
||||
class ApplyTest(TestCase):
|
||||
def _apply(self, per_disc_numbering=False, artist_credit=False):
|
||||
info = self.info
|
||||
94
test/autotag/test_match.py
Normal file
94
test/autotag/test_match.py
Normal file
|
|
@ -0,0 +1,94 @@
|
|||
import pytest
|
||||
|
||||
from beets.autotag.hooks import TrackInfo
|
||||
from beets.autotag.match import assign_items
|
||||
from beets.library import Item
|
||||
from beets.test.helper import ConfigMixin
|
||||
|
||||
|
||||
class TestAssignment(ConfigMixin):
|
||||
A = "one"
|
||||
B = "two"
|
||||
C = "three"
|
||||
|
||||
@pytest.fixture(scope="class", autouse=True)
|
||||
def _setup_config(self):
|
||||
self.config["match"]["track_length_grace"] = 10
|
||||
self.config["match"]["track_length_max"] = 30
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
# 'expected' is a tuple of expected (mapping, extra_items, extra_tracks)
|
||||
"item_titles, track_titles, expected",
|
||||
[
|
||||
# items ordering gets corrected
|
||||
([A, C, B], [A, B, C], ({A: A, B: B, C: C}, [], [])),
|
||||
# unmatched tracks are returned as 'extra_tracks'
|
||||
# the first track is unmatched
|
||||
([B, C], [A, B, C], ({B: B, C: C}, [], [A])),
|
||||
# the middle track is unmatched
|
||||
([A, C], [A, B, C], ({A: A, C: C}, [], [B])),
|
||||
# the last track is unmatched
|
||||
([A, B], [A, B, C], ({A: A, B: B}, [], [C])),
|
||||
# unmatched items are returned as 'extra_items'
|
||||
([A, C, B], [A, C], ({A: A, C: C}, [B], [])),
|
||||
],
|
||||
)
|
||||
def test_assign_tracks(self, item_titles, track_titles, expected):
|
||||
expected_mapping, expected_extra_items, expected_extra_tracks = expected
|
||||
|
||||
items = [Item(title=title) for title in item_titles]
|
||||
tracks = [TrackInfo(title=title) for title in track_titles]
|
||||
|
||||
item_info_pairs, extra_items, extra_tracks = assign_items(items, tracks)
|
||||
|
||||
assert (
|
||||
{i.title: t.title for i, t in item_info_pairs},
|
||||
[i.title for i in extra_items],
|
||||
[t.title for t in extra_tracks],
|
||||
) == (expected_mapping, expected_extra_items, expected_extra_tracks)
|
||||
|
||||
def test_order_works_when_track_names_are_entirely_wrong(self):
|
||||
# A real-world test case contributed by a user.
|
||||
def item(i, length):
|
||||
return Item(
|
||||
artist="ben harper",
|
||||
album="burn to shine",
|
||||
title=f"ben harper - Burn to Shine {i}",
|
||||
track=i,
|
||||
length=length,
|
||||
)
|
||||
|
||||
items = []
|
||||
items.append(item(1, 241.37243007106997))
|
||||
items.append(item(2, 342.27781704375036))
|
||||
items.append(item(3, 245.95070222338137))
|
||||
items.append(item(4, 472.87662515485437))
|
||||
items.append(item(5, 279.1759535763187))
|
||||
items.append(item(6, 270.33333768012))
|
||||
items.append(item(7, 247.83435613222923))
|
||||
items.append(item(8, 216.54504531525072))
|
||||
items.append(item(9, 225.72775379800484))
|
||||
items.append(item(10, 317.7643606963552))
|
||||
items.append(item(11, 243.57001238834192))
|
||||
items.append(item(12, 186.45916150485752))
|
||||
|
||||
def info(index, title, length):
|
||||
return TrackInfo(title=title, length=length, index=index)
|
||||
|
||||
trackinfo = []
|
||||
trackinfo.append(info(1, "Alone", 238.893))
|
||||
trackinfo.append(info(2, "The Woman in You", 341.44))
|
||||
trackinfo.append(info(3, "Less", 245.59999999999999))
|
||||
trackinfo.append(info(4, "Two Hands of a Prayer", 470.49299999999999))
|
||||
trackinfo.append(info(5, "Please Bleed", 277.86599999999999))
|
||||
trackinfo.append(info(6, "Suzie Blue", 269.30599999999998))
|
||||
trackinfo.append(info(7, "Steal My Kisses", 245.36000000000001))
|
||||
trackinfo.append(info(8, "Burn to Shine", 214.90600000000001))
|
||||
trackinfo.append(info(9, "Show Me a Little Shame", 224.0929999999999))
|
||||
trackinfo.append(info(10, "Forgiven", 317.19999999999999))
|
||||
trackinfo.append(info(11, "Beloved One", 243.733))
|
||||
trackinfo.append(info(12, "In the Lord's Arms", 186.13300000000001))
|
||||
|
||||
expected = list(zip(items, trackinfo)), [], []
|
||||
|
||||
assert assign_items(items, trackinfo) == expected
|
||||
Loading…
Reference in a new issue