Skip to content

Commit 6f51096

Browse files
committed
update
1 parent daba630 commit 6f51096

File tree

2 files changed

+5
-10
lines changed

2 files changed

+5
-10
lines changed

mllib/src/test/scala/org/apache/spark/ml/feature/LabeledPointSuite.scala

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,6 @@
1717

1818
package org.apache.spark.ml.feature
1919

20-
import scala.reflect.ClassTag
21-
2220
import org.apache.spark.{SparkConf, SparkFunSuite}
2321
import org.apache.spark.ml.linalg.Vectors
2422
import org.apache.spark.serializer.KryoSerializer

mllib/src/test/scala/org/apache/spark/mllib/regression/LabeledPointSuite.scala

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,6 @@
1717

1818
package org.apache.spark.mllib.regression
1919

20-
import scala.reflect.ClassTag
21-
2220
import org.apache.spark.{SparkConf, SparkFunSuite}
2321
import org.apache.spark.ml.feature.{LabeledPoint => NewLabeledPoint}
2422
import org.apache.spark.mllib.linalg.Vectors
@@ -63,13 +61,12 @@ class LabeledPointSuite extends SparkFunSuite {
6361

6462
val ser = new KryoSerializer(conf).newInstance()
6563

66-
def check[T: ClassTag](t: T) {
67-
assert(ser.deserialize[T](ser.serialize(t)) === t)
68-
}
69-
7064
val labeled1 = LabeledPoint(1.0, Vectors.dense(Array(1.0, 2.0)))
7165
val labeled2 = LabeledPoint(1.0, Vectors.sparse(10, Array(5, 7), Array(1.0, 2.0)))
72-
check(labeled1)
73-
check(labeled2)
66+
67+
Seq(labeled1, labeled2).foreach { l =>
68+
val l2 = ser.deserialize[LabeledPoint](ser.serialize(l))
69+
assert(l === l2)
70+
}
7471
}
7572
}

0 commit comments

Comments
 (0)