Skip to content

Commit 6416fa0

Browse files
author
Feynman Liang
committed
Add failing sparse matrix equals tests
1 parent 9897cc5 commit 6416fa0

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

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

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,22 @@ class MatricesSuite extends SparkFunSuite {
7474
}
7575
}
7676

77+
test("equals") {
78+
val dm1 = Matrices.dense(2, 2, Array(0.0, 1.0, 2.0, 3.0))
79+
assert(dm1 === dm1)
80+
assert(dm1 !== dm1.transpose)
81+
82+
val dm2 = Matrices.dense(2, 2, Array(0.0, 2.0, 1.0, 3.0))
83+
assert(dm1 === dm2.transpose)
84+
85+
val sm1 = dm1.toSparse
86+
assert(sm1 === sm1)
87+
assert(sm1 !== sm1.transpose)
88+
89+
val sm2 = dm2.toSparse
90+
assert(sm1 === sm2.transpose)
91+
}
92+
7793
test("matrix copies are deep copies") {
7894
val m = 3
7995
val n = 2

0 commit comments

Comments
 (0)