Skip to content
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.

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

Update names of transformer tests and files to match class names #284

Merged
merged 9 commits into from
Apr 16, 2019
Original file line number Diff line number Diff line change
@@ -42,7 +42,7 @@ import org.scalatest.{Assertions, FlatSpec, Matchers}


@RunWith(classOf[JUnitRunner])
class OpLdaTest extends FlatSpec with TestSparkContext {
class OpLDATest extends FlatSpec with TestSparkContext {

val inputData = Seq(
(0.0, Vectors.sparse(11, Array(0, 1, 2, 4, 5, 6, 7, 10), Array(1.0, 2.0, 6.0, 2.0, 3.0, 1.0, 1.0, 3.0))),
@@ -78,7 +78,7 @@ class OpLdaTest extends FlatSpec with TestSparkContext {
.toSeq
.map(_.getAs[Vector](0))

Spec[OpLdaTest] should "convert document term vectors into topic vectors" in {
Spec[OpLDA] should "convert document term vectors into topic vectors" in {
val f2Vec = new OpLDA().setInput(f2).setK(k).setSeed(seed).setMaxIter(maxIter)
val testTransformedData = f2Vec.fit(inputDS).transform(inputDS)
val output = f2Vec.getOutput()
Original file line number Diff line number Diff line change
@@ -45,7 +45,7 @@ import org.scalatest.junit.JUnitRunner


@RunWith(classOf[JUnitRunner])
class OpStandardScalerTest extends FlatSpec with TestSparkContext {
class OpScalarStandardScalerTest extends FlatSpec with TestSparkContext {
import spark.implicits._

// TODO: use TestFeatureBuilder instead
Original file line number Diff line number Diff line change
@@ -44,9 +44,9 @@ import org.slf4j.LoggerFactory


@RunWith(classOf[JUnitRunner])
class TextMapVectorizerTest extends FlatSpec with TestSparkContext with AttributeAsserts {
class TextMapPivotVectorizerTest extends FlatSpec with TestSparkContext with AttributeAsserts {

val log = LoggerFactory.getLogger(classOf[TextMapVectorizerTest])
val log = LoggerFactory.getLogger(classOf[TextMapPivotVectorizerTest])

lazy val (dataSet, top, bot) = TestFeatureBuilder("top", "bot",
Seq(
Original file line number Diff line number Diff line change
@@ -42,7 +42,7 @@ import org.scalatest.FlatSpec
import org.scalatest.junit.JUnitRunner

@RunWith(classOf[JUnitRunner])
class TransmogrifyTest extends FlatSpec with PassengerSparkFixtureTest with AttributeAsserts {
class TransmogrifierTest extends FlatSpec with PassengerSparkFixtureTest with AttributeAsserts {

val inputFeatures = Array[OPFeature](heightNoWindow, weight, gender)