From ab104f59019039eeec25e2b6b3310afdc381aee4 Mon Sep 17 00:00:00 2001
From: Arttu Voutilainen <avoutilainen@palantir.com>
Date: Thu, 21 Nov 2024 18:48:42 +0100
Subject: [PATCH] fix: add assert to check the pojo-proto-roundtrip

---
 .../test/scala/io/substrait/spark/SubstraitPlanTestBase.scala  | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/spark/src/test/scala/io/substrait/spark/SubstraitPlanTestBase.scala b/spark/src/test/scala/io/substrait/spark/SubstraitPlanTestBase.scala
index cbd7a151c..dea8f3a53 100644
--- 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