Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-23 09:03:49 +01:00
Code Issues Projects Releases Wiki Activity

lastgenre: add tests functions

Browse source
This commit is contained in:
Fabrice Laporte 2014-04-22 13:00:01 +02:00 committed by Adrian Sampson
parent 499a7b868e
commit 21f1607e8f
1 changed files with 49 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

61
test/test_lastgenre.py
View file

@ -18,27 +18,64 @@ from _common import unittest
from beetsplug import lastgenre
from beets import config
class LastGenrePluginTest(unittest.TestCase):
def setUp(self):
"""Set up configuration"""
lyrics.LastGenrePlugin()
lastgenre.LastGenrePlugin()
def _setup_config(self, whitelist=set(), branches=None, count=1):
config['lastgenre']['whitelist'] = whitelist
lastgenre.options['whitelist'] = whitelist
if branches:
config['lastgenre']['branches'] = branches
config['lastgenre']['c14n'] = True
lastgenre.options['branches'] = branches
lastgenre.options['c14n'] = True
else:
config['lastgenre']['c14n'] = False
lastgenre.options['c14n'] = False
config['lastgenre']['count'] = count
def test_c14n():
_setup_config(set('blues'),
[['blues'],
['blues', 'country blues'],
['blues', 'country blues', 'delta blues']])
def test_c14n(self):
"""Resolve genres that belong to a canonicalization branch.
"""
self._setup_config(whitelist=set(['blues']),
branches=[['blues'], ['blues', 'country blues'],
['blues', 'country blues',
'delta blues']])
self.assertEqual(lastgenre._strings_to_genre(['delta blues']),
'blues')
self.assertEqual(lastgenre._strings_to_genre(['delta blues']), 'Blues')
self.assertEqual(lastgenre._strings_to_genre(['rhytm n blues']), '')
def test_whitelist(self):
"""Keep only genres that are in the whitelist.
"""
self._setup_config(whitelist=set(['blues', 'rock', 'jazz']),
count=2)
self.assertEqual(lastgenre._strings_to_genre(['pop', 'blues']),
'Blues')
def test_count(self):
"""Keep the n first genres, as we expect them to be sorted from more to
less popular.
"""
self._setup_config(whitelist=set(['blues', 'rock', 'jazz']),
count=2)
self.assertEqual(lastgenre._strings_to_genre(
['jazz', 'pop', 'rock', 'blues']),
'Jazz, Rock')
def test_count_c14n(self):
"""Keep the n first genres, after having applied c14n when necessary
"""
self._setup_config(whitelist=set(['blues', 'rock', 'jazz']),
branches=[['blues'], ['blues', 'country blues']],
count=2)
self.assertEqual(lastgenre._strings_to_genre(
['jazz', 'pop', 'country blues', 'rock']),
'Jazz, Blues')
def suite():
return unittest.TestLoader().loadTestsFromName(__name__)
if __name__ == '__main__':
unittest.main(defaultTest='suite')