package com.github.mrpowers.spark.daria.sql

import org.apache.spark.sql.SparkSession

trait SparkSessionTestWrapper {

  lazy val spark: SparkSession = {
    SparkSession
      .builder()
      .master("local")
      .appName("spark session")
      .getOrCreate()
  }

}

package com.github.mrpowers.spark.daria.sql

import org.apache.spark.sql.SparkSession

trait SparkSessionTestWrapper {

  lazy val spark: SparkSession = {
    SparkSession
      .builder()
      .master("local")
      .appName("spark session")
      .config(
        "spark.sql.shuffle.partitions",
        "1"
      )
      .getOrCreate()
  }

}

Set shuffle partitions to 1 to make the test suite run faster
Set shuffle partitions to 1 to make the test suite run faster
