Make the points for a test be fewer

rxin · Jul 1, 2015 · 92c3efc · 92c3efc
1 parent c758692
commit 92c3efc
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
@@ -42,7 +42,7 @@ public void setUp() {
     sc = new JavaSparkContext("local", "JavaKMeansSuite");
     sql = new SQLContext(sc);
 
-    dataset = KMeansSuite.generateKMeansData(sql, 1000, 3, k);
+    dataset = KMeansSuite.generateKMeansData(sql, 50, 3, k);
   }
 
   @After

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
@@ -42,7 +42,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext {
   override def beforeAll(): Unit = {
     super.beforeAll()
 
-    dataset = KMeansSuite.generateKMeansData(sqlContext, 1000, 3, k)
+    dataset = KMeansSuite.generateKMeansData(sqlContext, 50, 3, k)
   }
 
   test("default parameters") {