Skip to content

Commit

Permalink
Minor refactor to UpdateSuiteBase
Browse files Browse the repository at this point in the history
GitOrigin-RevId: 0075439fc32d805a14f5c518b52626e3caa5f5d7
  • Loading branch information
sabir-akhadov authored and allisonport-db committed Sep 6, 2022
1 parent 24ef5df commit 0bedbbb
Showing 1 changed file with 7 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ import org.scalatest.BeforeAndAfterEach

import org.apache.spark.sql.{AnalysisException, DataFrame, QueryTest, Row}
import org.apache.spark.sql.execution.FileSourceScanExec
import org.apache.spark.sql.execution.datasources.FileFormat
import org.apache.spark.sql.internal.SQLConf
import org.apache.spark.sql.test.{SharedSparkSession, SQLTestUtils}
import org.apache.spark.sql.types._
Expand Down Expand Up @@ -671,8 +672,12 @@ abstract class UpdateSuiteBase
case f: FileSourceScanExec => f
})
// The first scan is for finding files to update. We only are matching against the key
// so that should be the only field in the schema
assert(scans.head.schema == StructType(Seq(StructField("key", IntegerType))))
// so that should be the only field in the schema.
assert(scans.head.schema == StructType(
Seq(
StructField("key", IntegerType)
)
))
}

/**
Expand Down

0 comments on commit 0bedbbb

Please sign in to comment.