600

longvu-db · Sep 23, 2024 · ebb0d4b · ebb0d4b
1 parent aadfb4f
commit ebb0d4b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/connector/connect/client/jvm/src/test/scala/org/apache/spark/sql/ClientE2ETestSuite.scala b/connector/connect/client/jvm/src/test/scala/org/apache/spark/sql/ClientE2ETestSuite.scala
@@ -228,16 +228,16 @@ class ClientE2ETestSuite
     assert(spark.conf.get("spark.connect.grpc.marshallerRecursionLimit").toInt == 4096)
     // spark.connect.grpc.marshallerRecursionLimit must be at least 2048, to handle certain
     // deep recursion cases.
-    for (a <- 1 to 700) {
+    for (a <- 1 to 605) {
       df = df.union(spark.range(a, a + 1))
     }
-    assert(df.collect().length == 701)
+    assert(df.collect().length == 606)
   }
 
   test("handle unknown exception") {
     var df = spark.range(1)
     val limit = spark.conf.get("spark.connect.grpc.marshallerRecursionLimit").toInt
-    for (a <- 1 to 1500) {
+    for (a <- 1 to 700) {
       df = df.union(spark.range(a, a + 1))
     }
     val ex = intercept[SparkException] {