Fix tests on Python 3

pybind · Aug 15, 2020 · 366f713 · 366f713
1 parent 64066d0
commit 366f713
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/tests/test_pytypes.py b/tests/test_pytypes.py
@@ -109,12 +109,16 @@ 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) 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)
+    else:
+        assert m.str_from_object(malformed_utf8) == "b'\\x80'"
+        assert m.str_from_handle(malformed_utf8) == "b'\\x80'"
 
 
 def test_bytes(doc):