diff --git a/unsafe/src/test/scala/org/apache/spark/sql/utils/SparkSessionTestWrapper.scala b/unsafe/src/test/scala/org/apache/spark/sql/utils/SparkSessionTestWrapper.scala
new file mode 100644
index 0000000..bc0706c
--- /dev/null
+++ b/unsafe/src/test/scala/org/apache/spark/sql/utils/SparkSessionTestWrapper.scala
@@ -0,0 +1,21 @@
+package org.apache.spark.sql.daria
+
+import org.apache.spark.sql.SparkSession
+
+trait SparkSessionTestWrapper {
+
+  lazy val spark: SparkSession = {
+    val session = SparkSession
+      .builder()
+      .master("local")
+      .appName("spark session")
+      .config(
+        "spark.sql.shuffle.partitions",
+        "1"
+      )
+      .getOrCreate()
+    session.sparkContext.setLogLevel("ERROR")
+    session
+  }
+
+}