diff --git a/test/test_importfeeds.py b/test/test_importfeeds.py index bee7fa28a..f9646c175 100644 --- a/test/test_importfeeds.py +++ b/test/test_importfeeds.py @@ -6,7 +6,7 @@ import shutil from _common import unittest from beets import config from beets.library import Item, Album, Library -from beetsplug.importfeeds import album_imported, ImportFeedsPlugin +from beetsplug.importfeeds import ImportFeedsPlugin class ImportfeedsTestTest(unittest.TestCase): @@ -30,7 +30,7 @@ class ImportfeedsTestTest(unittest.TestCase): self.lib.add(album) self.lib.add(item) - album_imported(self.lib, album) + self.importfeeds.album_imported(self.lib, album) playlist_path = os.path.join(self.feeds_dir, os.listdir(self.feeds_dir)[0]) self.assertTrue(playlist_path.endswith('album_name.m3u')) diff --git a/test/test_lastgenre.py b/test/test_lastgenre.py index 9aaace237..ee893be91 100644 --- a/test/test_lastgenre.py +++ b/test/test_lastgenre.py @@ -157,9 +157,10 @@ class LastGenrePluginTest(unittest.TestCase, TestHelper): tag2.item = MockPylastElem(u'Rap') return [tag1, tag2] - res = lastgenre._tags_for(MockPylastObj()) + plugin = lastgenre.LastGenrePlugin() + res = plugin._tags_for(MockPylastObj()) self.assertEqual(res, [u'pop', u'rap']) - res = lastgenre._tags_for(MockPylastObj(), min_weight=50) + res = plugin._tags_for(MockPylastObj(), min_weight=50) self.assertEqual(res, [u'pop']) def test_get_genre(self):