diff options
| author | Reynold Xin <rxin@databricks.com> | 2015-04-13 09:29:04 -0700 |
|---|---|---|
| committer | Reynold Xin <rxin@databricks.com> | 2015-04-13 09:29:04 -0700 |
| commit | c5b0b296b842926b5c07531a5affe8984bc799c5 (patch) | |
| tree | a01bcfeee05deb3c01436835a0263b84d9f03880 /mllib | |
| parent | 77620be76e82b6cdaae406cd752d3272656f5fe0 (diff) | |
| download | spark-c5b0b296b842926b5c07531a5affe8984bc799c5.tar.gz spark-c5b0b296b842926b5c07531a5affe8984bc799c5.tar.bz2 spark-c5b0b296b842926b5c07531a5affe8984bc799c5.zip | |
[SPARK-6765] Enable scalastyle on test code.
Turn scalastyle on for all test code. Most of the violations have been resolved in my previous pull requests:
Core: https://github.com/apache/spark/pull/5484
SQL: https://github.com/apache/spark/pull/5412
MLlib: https://github.com/apache/spark/pull/5411
GraphX: https://github.com/apache/spark/pull/5410
Streaming: https://github.com/apache/spark/pull/5409
Author: Reynold Xin <rxin@databricks.com>
Closes #5486 from rxin/test-style-enable and squashes the following commits:
01683de [Reynold Xin] Fixed new code.
a4ab46e [Reynold Xin] Fixed tests.
20adbc8 [Reynold Xin] Missed one violation.
5e36521 [Reynold Xin] [SPARK-6765] Enable scalastyle on test code.
Diffstat (limited to 'mllib')
| -rw-r--r-- | mllib/src/test/scala/org/apache/spark/ml/feature/VectorIndexerSuite.scala | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/VectorIndexerSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/VectorIndexerSuite.scala index 61c46c85a7..81ef831c42 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/feature/VectorIndexerSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/feature/VectorIndexerSuite.scala @@ -150,7 +150,8 @@ class VectorIndexerSuite extends FunSuite with MLlibTestSparkContext { val vectorIndexer = getIndexer.setMaxCategories(maxCategories) val model = vectorIndexer.fit(data) val categoryMaps = model.categoryMaps - assert(categoryMaps.keys.toSet === categoricalFeatures) // Chose correct categorical features + // Chose correct categorical features + assert(categoryMaps.keys.toSet === categoricalFeatures) val transformed = model.transform(data).select("indexed") val indexedRDD: RDD[Vector] = transformed.map(_.getAs[Vector](0)) val featureAttrs = AttributeGroup.fromStructField(transformed.schema("indexed")) |