diff --git a/python/pyspark/sql/tests/connect/test_connect_basic.py b/python/pyspark/sql/tests/connect/test_connect_basic.py index 9e7a5f2f4a546..ce1d4eeb298bc 100644 --- a/python/pyspark/sql/tests/connect/test_connect_basic.py +++ b/python/pyspark/sql/tests/connect/test_connect_basic.py @@ -387,7 +387,7 @@ def test_alias(self) -> None: with self.assertRaises(grpc.RpcError) as exc: self.connect.range(1, 10).select(col("id").alias("this", "is", "not")).collect() - self.assertIn("Buffer(this, is, not)", str(exc.exception)) + self.assertIn("(this, is, not)", str(exc.exception)) class ChannelBuilderTests(ReusedPySparkTestCase):