diff --git a/sql/core/src/test/scala/org/apache/spark/sql/PlanStabilitySuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/PlanStabilitySuite.scala index 76204c504c0ed..ca1b7ab7283c2 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/PlanStabilitySuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/PlanStabilitySuite.scala @@ -86,8 +86,9 @@ trait PlanStabilitySuite extends TPCDSBase with DisableAdaptiveExecutionSuite { private val regenerateGoldenFiles: Boolean = System.getenv("SPARK_GENERATE_GOLDEN_FILES") == "1" protected val baseResourcePath = { + val sparkHome = sys.props.getOrElse("spark.test.home", fail("spark.test.home is not set!")) // use the same way as `SQLQueryTestSuite` to get the resource path - java.nio.file.Paths.get("src", "test", "resources", "tpcds-plan-stability").toFile + java.nio.file.Paths.get(sparkHome, "sql", "core", "src", "test", "resources", "tpcds-plan-stability").toFile } private val referenceRegex = "#\\d+".r diff --git a/sql/core/src/test/scala/org/apache/spark/sql/TPCDSQueryTestSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/TPCDSQueryTestSuite.scala index 670048826e3bf..523cd6673f4c0 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/TPCDSQueryTestSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/TPCDSQueryTestSuite.scala @@ -79,7 +79,8 @@ class TPCDSQueryTestSuite extends QueryTest with TPCDSBase with SQLQueryTestHelp protected val baseResourcePath = { // use the same way as `SQLQueryTestSuite` to get the resource path - java.nio.file.Paths.get("src", "test", "resources", "tpcds-query-results") + val sparkHome = sys.props.getOrElse("spark.test.home", fail("spark.test.home is not set!")) + java.nio.file.Paths.get(sparkHome, "sql", "core", "src", "test", "resources", "tpcds-query-results") .toFile.getAbsolutePath }