diff --git a/src/bitmessageqt/tests/main.py b/src/bitmessageqt/tests/main.py index 9ffba909..9631878d 100644 --- a/src/bitmessageqt/tests/main.py +++ b/src/bitmessageqt/tests/main.py @@ -4,7 +4,8 @@ import Queue import sys import unittest -from qtpy import QtCore, QtWidgets +from PyQt5 import QtCore, QtWidgets +from six import string_types import bitmessageqt import queues @@ -39,10 +40,7 @@ class TestMain(unittest.TestCase): def test_translate(self): """Check the results of _translate() with various args""" - self.assertIsInstance( - _translate("MainWindow", "Test"), - unicode - ) + self.assertIsInstance(_translate("MainWindow", "Test"), string_types) class TestUISignaler(TestBase):