mirror of
https://github.com/beetbox/beets.git
synced 2025-12-09 18:12:19 +01:00
Removed unicode_literals from test_lastgenere
This commit is contained in:
parent
df0d51ea86
commit
d3d02e18b9
1 changed files with 24 additions and 25 deletions
|
|
@ -15,8 +15,7 @@
|
|||
|
||||
"""Tests for the 'lastgenre' plugin."""
|
||||
|
||||
from __future__ import (division, absolute_import, print_function,
|
||||
unicode_literals)
|
||||
from __future__ import (division, absolute_import, print_function)
|
||||
|
||||
from mock import Mock
|
||||
|
||||
|
|
@ -52,7 +51,7 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
"""
|
||||
self._setup_config()
|
||||
self.assertEqual(self.plugin._resolve_genres(['delta blues']),
|
||||
'Delta Blues')
|
||||
u'Delta Blues')
|
||||
|
||||
def test_c14n_only(self):
|
||||
"""Default c14n tree funnels up to most common genre except for *wrong*
|
||||
|
|
@ -60,16 +59,16 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
"""
|
||||
self._setup_config(canonical=True, count=99)
|
||||
self.assertEqual(self.plugin._resolve_genres(['delta blues']),
|
||||
'Blues')
|
||||
u'Blues')
|
||||
self.assertEqual(self.plugin._resolve_genres(['iota blues']),
|
||||
'Iota Blues')
|
||||
u'Iota Blues')
|
||||
|
||||
def test_whitelist_only(self):
|
||||
"""Default whitelist rejects *wrong* (non existing) genres.
|
||||
"""
|
||||
self._setup_config(whitelist=True)
|
||||
self.assertEqual(self.plugin._resolve_genres(['iota blues']),
|
||||
'')
|
||||
u'')
|
||||
|
||||
def test_whitelist_c14n(self):
|
||||
"""Default whitelist and c14n both activated result in all parents
|
||||
|
|
@ -77,7 +76,7 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
"""
|
||||
self._setup_config(canonical=True, whitelist=True, count=99)
|
||||
self.assertEqual(self.plugin._resolve_genres(['delta blues']),
|
||||
'Delta Blues, Blues')
|
||||
u'Delta Blues, Blues')
|
||||
|
||||
def test_whitelist_custom(self):
|
||||
"""Keep only genres that are in the whitelist.
|
||||
|
|
@ -85,11 +84,11 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
self._setup_config(whitelist=set(['blues', 'rock', 'jazz']),
|
||||
count=2)
|
||||
self.assertEqual(self.plugin._resolve_genres(['pop', 'blues']),
|
||||
'Blues')
|
||||
u'Blues')
|
||||
|
||||
self._setup_config(canonical='', whitelist=set(['rock']))
|
||||
self.assertEqual(self.plugin._resolve_genres(['delta blues']),
|
||||
'')
|
||||
u'')
|
||||
|
||||
def test_count(self):
|
||||
"""Keep the n first genres, as we expect them to be sorted from more to
|
||||
|
|
@ -99,7 +98,7 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
count=2)
|
||||
self.assertEqual(self.plugin._resolve_genres(
|
||||
['jazz', 'pop', 'rock', 'blues']),
|
||||
'Jazz, Rock')
|
||||
u'Jazz, Rock')
|
||||
|
||||
def test_count_c14n(self):
|
||||
"""Keep the n first genres, after having applied c14n when necessary
|
||||
|
|
@ -111,14 +110,14 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
# second slot
|
||||
self.assertEqual(self.plugin._resolve_genres(
|
||||
['jazz', 'pop', 'country blues', 'rock']),
|
||||
'Jazz, Blues')
|
||||
u'Jazz, Blues')
|
||||
|
||||
def test_c14n_whitelist(self):
|
||||
"""Genres first pass through c14n and are then filtered
|
||||
"""
|
||||
self._setup_config(canonical=True, whitelist=set(['rock']))
|
||||
self.assertEqual(self.plugin._resolve_genres(['delta blues']),
|
||||
'')
|
||||
u'')
|
||||
|
||||
def test_empty_string_enables_canonical(self):
|
||||
"""For backwards compatibility, setting the `canonical` option
|
||||
|
|
@ -126,7 +125,7 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
"""
|
||||
self._setup_config(canonical='', count=99)
|
||||
self.assertEqual(self.plugin._resolve_genres(['delta blues']),
|
||||
'Blues')
|
||||
u'Blues')
|
||||
|
||||
def test_empty_string_enables_whitelist(self):
|
||||
"""Again for backwards compatibility, setting the `whitelist`
|
||||
|
|
@ -134,14 +133,14 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
"""
|
||||
self._setup_config(whitelist='')
|
||||
self.assertEqual(self.plugin._resolve_genres(['iota blues']),
|
||||
'')
|
||||
u'')
|
||||
|
||||
def test_no_duplicate(self):
|
||||
"""Remove duplicated genres.
|
||||
"""
|
||||
self._setup_config(count=99)
|
||||
self.assertEqual(self.plugin._resolve_genres(['blues', 'blues']),
|
||||
'Blues')
|
||||
u'Blues')
|
||||
|
||||
def test_tags_for(self):
|
||||
class MockPylastElem(object):
|
||||
|
|
@ -189,29 +188,29 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper):
|
|||
|
||||
config['lastgenre'] = {'force': False}
|
||||
res = self.plugin._get_genre(item)
|
||||
self.assertEqual(res, (item.genre, 'keep'))
|
||||
self.assertEqual(res, (item.genre, u'keep'))
|
||||
|
||||
config['lastgenre'] = {'force': True, 'source': 'track'}
|
||||
config['lastgenre'] = {'force': True, 'source': u'track'}
|
||||
res = self.plugin._get_genre(item)
|
||||
self.assertEqual(res, (MOCK_GENRES['track'], 'track'))
|
||||
self.assertEqual(res, (MOCK_GENRES['track'], u'track'))
|
||||
|
||||
config['lastgenre'] = {'source': 'album'}
|
||||
config['lastgenre'] = {'source': u'album'}
|
||||
res = self.plugin._get_genre(item)
|
||||
self.assertEqual(res, (MOCK_GENRES['album'], 'album'))
|
||||
self.assertEqual(res, (MOCK_GENRES['album'], u'album'))
|
||||
|
||||
config['lastgenre'] = {'source': 'artist'}
|
||||
config['lastgenre'] = {'source': u'artist'}
|
||||
res = self.plugin._get_genre(item)
|
||||
self.assertEqual(res, (MOCK_GENRES['artist'], 'artist'))
|
||||
self.assertEqual(res, (MOCK_GENRES['artist'], u'artist'))
|
||||
|
||||
MOCK_GENRES['artist'] = None
|
||||
res = self.plugin._get_genre(item)
|
||||
self.assertEqual(res, (item.genre, 'original'))
|
||||
self.assertEqual(res, (item.genre, u'original'))
|
||||
|
||||
config['lastgenre'] = {'fallback': 'rap'}
|
||||
config['lastgenre'] = {'fallback': u'rap'}
|
||||
item.genre = None
|
||||
res = self.plugin._get_genre(item)
|
||||
self.assertEqual(res, (config['lastgenre']['fallback'].get(),
|
||||
'fallback'))
|
||||
u'fallback'))
|
||||
|
||||
|
||||
def suite():
|
||||
|
|
|
|||
Loading…
Reference in a new issue