Fixed a few python2 tests.

2016-07-26 01:15:18 -07:00
parent 5a5d9ac56d
commit 620434eed0
3 changed files with 8 additions and 5 deletions
--- a/tests/test_terminal.py
+++ b/tests/test_terminal.py
@@ -154,7 +154,7 @@ def test_terminal_clean_unescape_html(terminal, use_ascii):
    terminal.config['ascii'] = use_ascii
    text = terminal.clean('&lt;')
    assert isinstance(text, six.binary_type)
-    assert text.decode('ascii' if ascii else 'utf-8') == '<'
+    assert text.decode('ascii' if use_ascii else 'utf-8') == '<'


@pytest.mark.parametrize('use_ascii', [True, False])