From 6906adc6b50a70f1e4e0f7c3384bb9c32c9cdcce Mon Sep 17 00:00:00 2001
From: Bing Li <bing.li@snowflake.com>
Date: Fri, 27 Sep 2024 14:01:18 -0700
Subject: [PATCH] fix udf test

---
 src/test/scala/com/snowflake/snowpark/TestUtils.scala | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/test/scala/com/snowflake/snowpark/TestUtils.scala b/src/test/scala/com/snowflake/snowpark/TestUtils.scala
index 46b22227..0221fa10 100644
--- a/src/test/scala/com/snowflake/snowpark/TestUtils.scala
+++ b/src/test/scala/com/snowflake/snowpark/TestUtils.scala
@@ -29,6 +29,7 @@ import net.snowflake.client.jdbc.{
   SnowflakeConnectString,
   SnowflakeConnectionV1
 }
+import org.scalatest.funsuite.AnyFunSuite
 
 import java.security.Provider
 import scala.collection.JavaConverters._
@@ -142,6 +143,8 @@ object TestUtils extends Logging {
 
     List(
       classOf[BeforeAndAfterAll], // scala test jar
+      classOf[AnyFunSuite],
+      classOf[org.scalatest.compatible.Assertion],
       classOf[org.scalactic.TripleEquals], // scalactic jar
       classOf[io.opentelemetry.exporters.inmemory.InMemorySpanExporter],
       classOf[io.opentelemetry.sdk.trace.export.SpanExporter])