Add a second b entry so b gets 0 for sure

holdenk · holdenk · commit 045bf22ab185 · 2015-08-06T10:44:29.000-07:00
diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala
@@ -51,7 +51,7 @@ class StringIndexerSuite extends SparkFunSuite with MLlibTestSparkContext {
   }
 
   test("StringIndexerUnseen") {
-    val data = sc.parallelize(Seq((0, "a"), (1, "b")), 2)
+    val data = sc.parallelize(Seq((0, "a"), (1, "b"), (4, "b")), 2)
     val data2 = sc.parallelize(Seq((0, "a"), (1, "b"), (2, "c")), 2)
     val df = sqlContext.createDataFrame(data).toDF("id", "label")
     val df2 = sqlContext.createDataFrame(data2).toDF("id", "label")

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ class StringIndexerSuite extends SparkFunSuite with MLlibTestSparkContext {`
`51`	`51`	`}`
`52`	`52`
`53`	`53`	`test("StringIndexerUnseen") {`
`54`		`- val data = sc.parallelize(Seq((0, "a"), (1, "b")), 2)`
	`54`	`+ val data = sc.parallelize(Seq((0, "a"), (1, "b"), (4, "b")), 2)`
`55`	`55`	`val data2 = sc.parallelize(Seq((0, "a"), (1, "b"), (2, "c")), 2)`
`56`	`56`	`val df = sqlContext.createDataFrame(data).toDF("id", "label")`
`57`	`57`	`val df2 = sqlContext.createDataFrame(data2).toDF("id", "label")`