Skip to content

Commit a2c38b9

Browse files
committed
Handle Spark 3
1 parent 05a4aa3 commit a2c38b9

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

Tests/SparkConnectTests/CatalogTests.swift

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,11 @@ struct CatalogTests {
152152
let path = "/tmp/\(tableName)"
153153
try await SQLHelper.withTable(spark, tableName)({
154154
try await spark.range(2).write.orc(path)
155-
let expected = [Row("id", nil, "bigint", true, false, false, false)]
155+
let expected = if await spark.version.starts(with: "4.") {
156+
[Row("id", nil, "bigint", true, false, false, false)]
157+
} else {
158+
[Row("id", nil, "bigint", true, false, false)]
159+
}
156160
#expect(try await spark.catalog.createTable(tableName, path, source: "orc").count() == 2)
157161
#expect(try await spark.catalog.listColumns(tableName).collect() == expected)
158162
#expect(try await spark.catalog.listColumns("default.\(tableName)").collect() == expected)
@@ -162,7 +166,11 @@ struct CatalogTests {
162166
let viewName = "VIEW_" + UUID().uuidString.replacingOccurrences(of: "-", with: "")
163167
try await SQLHelper.withTempView(spark, viewName)({
164168
try await spark.range(1).createTempView(viewName)
165-
let expected = [Row("id", nil, "bigint", false, false, false, false)]
169+
let expected = if await spark.version.starts(with: "4.") {
170+
[Row("id", nil, "bigint", false, false, false, false)]
171+
} else {
172+
[Row("id", nil, "bigint", false, false, false)]
173+
}
166174
#expect(try await spark.catalog.listColumns(viewName).collect() == expected)
167175
})
168176

0 commit comments

Comments
 (0)