Skip to content

Commit

Permalink
Fixes #82
Browse files Browse the repository at this point in the history
  • Loading branch information
TebaleloS committed Mar 15, 2023
1 parent d1809bf commit adf1baf
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ package za.co.absa.spark.commons.sql

import org.apache.spark.sql.Column
import org.apache.spark.sql.functions.col
//import za.co.absa.spark.commons.utils.SchemaUtils
import za.co.absa.spark.commons.utils.SchemaUtils

import scala.util.{Success, Try}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,18 +29,17 @@ class OncePerSparkSessionTest extends AnyFunSuite with MockitoSugar with SparkTe
val anotherSpark: SparkSession = mock[SparkSession]
class UDFLibraryA()(implicit sparkToRegisterTo: SparkSession) extends OncePerSparkSession() {
this.register(sparkToRegisterTo)
override protected def registerBody(implicit spark: SparkSession): Unit = {
override protected def registerBody(spark: SparkSession): Unit = {
libraryAInitCounter += 1
}
}

class UDFLibraryB()(implicit sparkToRegisterTo: SparkSession) extends OncePerSparkSession(sparkToRegisterTo) {
override protected def registerBody(implicit spark: SparkSession): Unit = {
override protected def registerBody(spark: SparkSession): Unit = {
libraryBInitCounter += 1
}
}


new UDFLibraryA()
new UDFLibraryA()
new UDFLibraryB()
Expand Down

0 comments on commit adf1baf

Please sign in to comment.