Skip to content

[SPARK-8446] [SQL] Add helper functions for testing SparkPlan physical operators #6885

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
JoshRosen wants to merge 8 commits into from
Closed

[SPARK-8446] [SQL] Add helper functions for testing SparkPlan physical operators #6885

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
a46144a
WIP
JoshRosen Jun 17, 2015
996332a
Add types so that tests compile
JoshRosen Jun 18, 2015
c60a44d
Manually bind references
JoshRosen Jun 18, 2015
d9ab1e4
Add simple resolver
marmbrus Jun 18, 2015
a80f9b0
Merge pull request #4 from marmbrus/pr/6885
JoshRosen Jun 18, 2015
ae1896b
Provide implicits automatically
JoshRosen Jun 18, 2015
84214be
Add an extra column which isn't part of the sort
JoshRosen Jun 18, 2015
f8ce275
Fix some IntelliJ inspections and delete some dead code
JoshRosen Jun 18, 2015
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Provide implicits automatically
  • Loading branch information
@JoshRosen
JoshRosen committed Jun 18, 2015
commit ae1896bdeea9504494b539348ae2467d2906e900
5 changes: 0 additions & 5 deletions sql/core/src/test/scala/org/apache/spark/sql/execution/SortSuite.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,14 +17,9 @@

package org.apache.spark.sql.execution

import org.apache.spark.sql.catalyst.expressions.{BoundReference, Ascending, SortOrder}
import org.apache.spark.sql.catalyst.dsl.expressions._

import org.apache.spark.sql.test.TestSQLContext
import org.apache.spark.sql.types.{IntegerType, StringType}

class SortSuite extends SparkPlanTest {
import TestSQLContext.implicits.localSeqToDataFrameHolder

test("basic sorting using ExternalSort") {

Expand Down
12 changes: 10 additions & 2 deletions sql/core/src/test/scala/org/apache/spark/sql/execution/SparkPlanTest.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,9 @@

package org.apache.spark.sql.execution

import scala.util.control.NonFatal
import scala.language.implicitConversions
import scala.reflect.runtime.universe.TypeTag
import scala.util.control.NonFatal

import org.apache.spark.SparkFunSuite

Expand All @@ -27,14 +28,21 @@ import org.apache.spark.sql.catalyst.expressions.BoundReference
import org.apache.spark.sql.catalyst.util._

import org.apache.spark.sql.test.TestSQLContext
import org.apache.spark.sql.{Row, DataFrame}
import org.apache.spark.sql.{DataFrameHolder, Row, DataFrame}

/**
* Base class for writing tests for individual physical operators. For an example of how this
* class's test helper methods can be used, see [[SortSuite]].
*/
class SparkPlanTest extends SparkFunSuite {

/**
* Creates a DataFrame from a local Seq of Product.
*/
implicit def localSeqToDataFrameHolder[A <: Product : TypeTag](data: Seq[A]): DataFrameHolder = {
TestSQLContext.implicits.localSeqToDataFrameHolder(data)
}

/**
* Runs the plan and makes sure the answer matches the expected result.
* @param input the input data to be used.
Expand Down