diff --git a/tests/termstr_test.py b/tests/termstr_test.py index 5dffe68..0721176 100755 --- a/tests/termstr_test.py +++ b/tests/termstr_test.py @@ -8,7 +8,6 @@ import unittest os.environ["TERM"] = "xterm-256color" import fill3.terminal as terminal -from termstr import TermStr, CharStyle import termstr @@ -46,7 +45,7 @@ class TermStrTests(unittest.TestCase): bold_style = termstr.CharStyle(3, 5, is_bold=True) foo_bold = termstr.TermStr("foo", bold_style) self.assertEqual(repr(foo_bold), "") - self.assertEqual(foo + "bar", TermStr("foobar")) + self.assertEqual(foo + "bar", termstr.TermStr("foobar")) self.assertEqual(foo + termstr.TermStr("bar"), termstr.TermStr("foobar")) self.assertEqual("bar" + foo, termstr.TermStr("barfoo"))