diff --git a/python/pyspark/sql/tests/test_arrow.py b/python/pyspark/sql/tests/test_arrow.py index 1f96d2cdd9cff..f5330835a3f22 100644 --- a/python/pyspark/sql/tests/test_arrow.py +++ b/python/pyspark/sql/tests/test_arrow.py @@ -271,7 +271,7 @@ def test_createDataFrame_with_incorrect_schema(self): fields[0], fields[1] = fields[1], fields[0] # swap str with int wrong_schema = StructType(fields) with QuietTest(self.sc): - with self.assertRaisesRegexp(Exception, "integer.*required.*got.*str"): + with self.assertRaisesRegexp(Exception, "integer.*required"): self.spark.createDataFrame(pdf, schema=wrong_schema) def test_createDataFrame_with_names(self):