Bumping up test matrix size to eliminate random failures

2 files changed, 12 insertions, 12 deletions

diff --git a/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java b/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java
index c8e59f7399..eafee060cd 100644
--- a/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java
+++ b/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java
@@ -102,8 +102,8 @@ public class JavaALSSuite implements Serializable {
   public void runALSUsingStaticMethods() {
     int features = 1;
     int iterations = 15;
-    int users = 10;
-    int products = 10;
+    int users = 50;
+    int products = 100;
     scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList(
         users, products, features, 0.7, false);
 
@@ -116,8 +116,8 @@ public class JavaALSSuite implements Serializable {
   public void runALSUsingConstructor() {
     int features = 2;
     int iterations = 15;
-    int users = 20;
-    int products = 30;
+    int users = 100;
+    int products = 200;
     scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList(
         users, products, features, 0.7, false);
 
@@ -133,8 +133,8 @@ public class JavaALSSuite implements Serializable {
   public void runImplicitALSUsingStaticMethods() {
     int features = 1;
     int iterations = 15;
-    int users = 40;
-    int products = 80;
+    int users = 80;
+    int products = 160;
     scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList(
       users, products, features, 0.7, true);
 
@@ -147,8 +147,8 @@ public class JavaALSSuite implements Serializable {
   public void runImplicitALSUsingConstructor() {
     int features = 2;
     int iterations = 15;
-    int users = 50;
-    int products = 100;
+    int users = 100;
+    int products = 200;
     scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList(
       users, products, features, 0.7, true);
 
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala
index 1ab181d35a..fafc5ec5f2 100644
--- a/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala
+++ b/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala
@@ -87,19 +87,19 @@ class ALSSuite extends FunSuite with BeforeAndAfterAll {
   }
 
   test("rank-1 matrices") {
-    testALS(10, 20, 1, 15, 0.7, 0.3)
+    testALS(50, 100, 1, 15, 0.7, 0.3)
   }
 
   test("rank-2 matrices") {
-    testALS(20, 30, 2, 15, 0.7, 0.3)
+    testALS(100, 200, 2, 15, 0.7, 0.3)
   }
 
   test("rank-1 matrices implicit") {
-    testALS(40, 80, 1, 15, 0.7, 0.4, true)
+    testALS(80, 160, 1, 15, 0.7, 0.4, true)
   }
 
   test("rank-2 matrices implicit") {
-    testALS(50, 100, 2, 15, 0.7, 0.4, true)
+    testALS(100, 200, 2, 15, 0.7, 0.4, true)
   }
 
   /**