Skip to content

Commit 75dced3

Browse files
committed
update test case
1 parent d129a66 commit 75dced3

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

mllib/src/test/scala/org/apache/spark/mllib/linalg/BreezeVectorConversionSuite.scala

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -56,15 +56,12 @@ class BreezeVectorConversionSuite extends FunSuite {
5656
assert(vec.values.eq(values), "should not copy data")
5757
}
5858

59-
test("sparse breeze by vector builder to vector") {
60-
val builder = new BVB[Double](n)
61-
for (i <- 0 until indices.length) {
62-
builder.add(indices(i), values(i))
63-
}
64-
val breeze = builder.toSparseVector
59+
test("sparse breeze with partially-used arrays to vector") {
60+
val activeSize = 3
61+
val breeze = new BSV[Double](indices, values, activeSize, n)
6562
val vec = Vectors.fromBreeze(breeze).asInstanceOf[SparseVector]
6663
assert(vec.size === n)
67-
assert(vec.indices === indices)
68-
assert(vec.values === values)
64+
assert(vec.indices === indices.slice(0, activeSize))
65+
assert(vec.values === values.slice(0, activeSize))
6966
}
7067
}

0 commit comments

Comments
 (0)