From 92c3efc4f60f88fb2a7c8f6cf67435868e46f746 Mon Sep 17 00:00:00 2001
From: Yu ISHIKAWA <yuu.ishikawa@gmail.com>
Date: Wed, 1 Jul 2015 12:02:58 +0900
Subject: [PATCH] Make the points for a test be fewer

---
 .../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 a2c3d79c78cd0..d5a29f3ce4c86 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
@@ -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
index a293525854e0b..04d7b5946cbcf 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
@@ -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") {