Fix tests on Python 3

pybind · Aug 16, 2020 · f7d376c · f7d376c
1 parent d073cf1
commit f7d376c
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/tests/test_pytypes.py b/tests/test_pytypes.py
@@ -111,12 +111,14 @@ def __repr__(self):
     assert s1 == s2
 
     malformed_utf8 = b"\x80"
-    with pytest.raises(UnicodeDecodeError) as excinfo:
-        assert m.str_from_object(malformed_utf8)
-    assert 'invalid start byte' in str(excinfo.value)
-    with pytest.raises(UnicodeDecodeError) as excinfo:
-        assert m.str_from_handle(malformed_utf8)
-    assert 'invalid start byte' in str(excinfo.value)
+    if sys.version_info.major < 3:
+        with pytest.raises(UnicodeDecodeError):
+            assert m.str_from_object(malformed_utf8)
+        with pytest.raises(UnicodeDecodeError):
+            assert m.str_from_handle(malformed_utf8)
+    else:
+        assert m.str_from_object(malformed_utf8) == "b'\\x80'"
+        assert m.str_from_handle(malformed_utf8) == "b'\\x80'"
 
 
 def test_bytes(doc):