fix: add assert to check the pojo-proto-roundtrip

substrait-io · Nov 21, 2024 · ab104f5 · ab104f5
1 parent e374c85
commit ab104f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spark/src/test/scala/io/substrait/spark/SubstraitPlanTestBase.scala b/spark/src/test/scala/io/substrait/spark/SubstraitPlanTestBase.scala
@@ -95,7 +95,8 @@ trait SubstraitPlanTestBase { self: SharedSparkSession =>
 
     val extensionCollector = new ExtensionCollector;
     val proto = new RelProtoConverter(extensionCollector).toProto(pojoRel)
-    new ProtoRelConverter(extensionCollector, SparkExtension.COLLECTION).from(proto)
+    val pojoFromProto = new ProtoRelConverter(extensionCollector, SparkExtension.COLLECTION).from(proto)
+    assertResult(pojoRel)(pojoFromProto)
 
     pojoRel2.shouldEqualPlainly(pojoRel)
     logicalPlan2