From 0613cb7d138cdb61f86f199bf52678d70befbaf0 Mon Sep 17 00:00:00 2001 From: Fabrice Laporte Date: Sun, 27 Apr 2014 22:31:31 +0200 Subject: [PATCH] Update _setup_config Give an empty string to obtain default whitelist, None for no whitelist or a stringlist for a custom one. --- test/test_lastgenre.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_lastgenre.py b/test/test_lastgenre.py index b4fa9cc4d..fb2952799 100644 --- a/test/test_lastgenre.py +++ b/test/test_lastgenre.py @@ -27,11 +27,11 @@ lastGenrePlugin = lastgenre.LastGenrePlugin() class LastGenrePluginTest(unittest.TestCase): - def _setup_config(self, whitelist=set(), canonical=None, count=1): + def _setup_config(self, whitelist=None, canonical=None, count=1): config['lastgenre']['canonical'] = canonical config['lastgenre']['count'] = count - config['lastgenre']['whitelist'] = \ - os.path.join(os.path.dirname(lastgenre.__file__), 'genres.txt') + if whitelist == '': + config['lastgenre']['whitelist'] = whitelist lastGenrePlugin.setup() if whitelist: lastGenrePlugin.whitelist = whitelist