Add assertions

feynmanliang · Jul 17, 2015 · be752de · be752de
1 parent a14939b
commit be752de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mllib/src/test/java/org/apache/spark/ml/clustering/JavaKMeansSuite.java b/mllib/src/test/java/org/apache/spark/ml/clustering/JavaKMeansSuite.java
@@ -66,7 +66,7 @@ public void fitAndTransform() {
     List<String> columns = Arrays.asList(transformed.columns());
     List<String> expectedColumns = Arrays.asList("features", "prediction");
     for (String column: expectedColumns) {
-      columns.contains(column);
+      assertTrue(columns.contains(column));
     }
   }
 }
diff --git a/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala
@@ -105,7 +105,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext {
     val transformed = model.transform(dataset)
     val expectedColumns = Array("features", predictionColName)
     expectedColumns.foreach { column =>
-      transformed.columns.contains(column)
+      assert(transformed.columns.contains(column))
     }
     val clusters = transformed.select(predictionColName).map(_.getInt(0)).distinct().collect().toSet
     assert(clusters.size === k)