From be752de88b45a43da8609517796b00633f943e79 Mon Sep 17 00:00:00 2001 From: Yu ISHIKAWA Date: Sat, 18 Jul 2015 07:35:38 +0900 Subject: [PATCH] Add assertions --- .../java/org/apache/spark/ml/clustering/JavaKMeansSuite.java | 2 +- .../test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala | 2 +- 2 files changed, 2 insertions(+), 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 index 5c9c0c1382c8b..d09fa7fd5637c 100644 --- 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 columns = Arrays.asList(transformed.columns()); List 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 index 1649e24442830..1f15ac02f4008 100644 --- 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)