diff --git a/tests/test_collection.py b/tests/test_collection.py index 594e8cb..57a8c3d 100644 --- a/tests/test_collection.py +++ b/tests/test_collection.py @@ -188,7 +188,13 @@ class SetIntSet: with self.assertRaises(TypeError) as err_info: schema.load({"value": {set()}}) - self.assertEqual(str(err_info.exception), "unhashable type: 'set'") + self.assertIn( + str(err_info.exception), + { + "unhashable type: 'set'", + "cannot use 'set' as a set element (unhashable type: 'set')", + }, + ) @dataclass() class Elm: @@ -202,7 +208,13 @@ class SetOfDataclass: with self.assertRaises(TypeError) as err_info: schema.load({"value": {{"value": {set()}}}}) - self.assertEqual(str(err_info.exception), "unhashable type: 'set'") + self.assertIn( + str(err_info.exception), + { + "unhashable type: 'set'", + "cannot use 'set' as a set element (unhashable type: 'set')", + }, + ) def test_set_of_frozen_dataclass(self): @dataclass(frozen=True)