diff --git a/test/test_ui.py b/test/test_ui.py index 519962b41..8c93a83ea 100644 --- a/test/test_ui.py +++ b/test/test_ui.py @@ -737,7 +737,7 @@ class UpdateTest(IOMixin, BeetsTestCase): assert item.lyrics != "new lyrics" -class PrintTest(IOMixin, BeetsTestCase): +class PrintTest(IOMixin, unittest.TestCase): def test_print_without_locale(self): lang = os.environ.get("LANG") if lang: @@ -1111,7 +1111,7 @@ class ConfigTest(TestPluginTestCase): ) -class ShowModelChangeTest(IOMixin, BeetsTestCase): +class ShowModelChangeTest(IOMixin, unittest.TestCase): def setUp(self): super().setUp() self.a = _common.item() @@ -1162,7 +1162,7 @@ class ShowModelChangeTest(IOMixin, BeetsTestCase): assert "bar" in out -class ShowChangeTest(IOMixin, BeetsTestCase): +class ShowChangeTest(IOMixin, unittest.TestCase): def setUp(self): super().setUp() diff --git a/test/test_ui_init.py b/test/test_ui_init.py index 0f42d7258..f6c9fe245 100644 --- a/test/test_ui_init.py +++ b/test/test_ui_init.py @@ -16,6 +16,7 @@ import os import shutil +import unittest from copy import deepcopy from random import random @@ -24,7 +25,7 @@ from beets.test import _common from beets.test.helper import BeetsTestCase, IOMixin, control_stdin -class InputMethodsTest(IOMixin, BeetsTestCase): +class InputMethodsTest(IOMixin, unittest.TestCase): def _print_helper(self, s): print(s)