Fix tests on Python 3

pybind · Aug 17, 2020 · 4fc3b1b · 4fc3b1b
1 parent bd8f366
commit 4fc3b1b
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 env.PY2:
+        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):