.

sabir-akhadov · May 21, 2024 · bd92170 · bd92170
1 parent cdd11f8
commit bd92170
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...c/test/scala/org/apache/spark/sql/delta/columnmapping/RemoveColumnMappingSuiteUtils.scala b/...c/test/scala/org/apache/spark/sql/delta/columnmapping/RemoveColumnMappingSuiteUtils.scala
@@ -102,8 +102,11 @@ trait RemoveColumnMappingSuiteUtils extends QueryTest with DeltaColumnMappingSui
       spark.table(tableName = testTableName).select(logicalColumnName),
       originalData)
     val newSnapshot = deltaLog.update()
-    // Drop feature adds 2 empty commits.
-    val versionsAddedByRewrite = if (droppedFeature) 3 else 1
+    val versionsAddedByRewrite = if (droppedFeature) {
+      2
+    } else {
+      1
+    }
     assert(newSnapshot.version - startingVersion == versionsAddedByRewrite,
       s"Should rewrite the table in $versionsAddedByRewrite commits.")