diff --git a/test/test_ui_importer.py b/test/test_ui_importer.py index 4f0de1c92..3248b25a3 100644 --- a/test/test_ui_importer.py +++ b/test/test_ui_importer.py @@ -19,8 +19,7 @@ test_importer module. But here the test importer inherits from ``TerminalImportSession``. So we test this class, too. """ -from __future__ import (division, absolute_import, print_function, - unicode_literals) +from __future__ import (division, absolute_import, print_function) from test._common import unittest, DummyIO from test import test_importer @@ -59,21 +58,21 @@ class TestTerminalImportSession(TerminalImportSession): choice = self.default_choice if choice == importer.action.APPLY: - self.io.addinput('A') + self.io.addinput(u'A') elif choice == importer.action.ASIS: - self.io.addinput('U') + self.io.addinput(u'U') elif choice == importer.action.ALBUMS: - self.io.addinput('G') + self.io.addinput(u'G') elif choice == importer.action.TRACKS: - self.io.addinput('T') + self.io.addinput(u'T') elif choice == importer.action.SKIP: - self.io.addinput('S') + self.io.addinput(u'S') elif isinstance(choice, int): - self.io.addinput('M') + self.io.addinput(u'M') self.io.addinput(unicode(choice)) self._add_choice_input() else: - raise Exception('Unknown choice %s' % choice) + raise Exception(u'Unknown choice %s' % choice) class TerminalImportSessionSetup(object):