Fix schema assertion for Union join

Ackuq · Mar 14, 2022 · 077be2d · 077be2d
1 parent a13dbda
commit 077be2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scala/src/test/scala/data/SmallDataUnion.scala b/scala/src/test/scala/data/SmallDataUnion.scala
@@ -57,7 +57,7 @@ class SmallDataUnion(spark: SparkSession) extends SmallData(spark) {
   )
   private val PIT_2_schema: StructType = StructType(
     Seq(
-      StructField("id", IntegerType, nullable = true),
+      StructField("id", IntegerType, nullable = false),
       StructField("fg1_ts", IntegerType, nullable = true),
       StructField("fg1_value", StringType, nullable = true),
       StructField("fg2_ts", IntegerType, nullable = true),
@@ -66,7 +66,7 @@ class SmallDataUnion(spark: SparkSession) extends SmallData(spark) {
   )
   private val PIT_3_schema: StructType = StructType(
     Seq(
-      StructField("id", IntegerType, nullable = true),
+      StructField("id", IntegerType, nullable = false),
       StructField("fg1_ts", IntegerType, nullable = true),
       StructField("fg1_value", StringType, nullable = true),
       StructField("fg2_ts", IntegerType, nullable = true),