Skip to content

Commit 34d5853

Browse files
committed
refactored tests to make tests succeed
1 parent 4bd1103 commit 34d5853

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mllib/src/test/scala/org/apache/spark/mllib/clustering/LDASuite.scala

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -118,10 +118,10 @@ class LDASuite extends FunSuite with MLlibTestSparkContext {
118118
val docVertexIds = docIds
119119
val termIds = Array(0, 1, 2)
120120
val termVertexIds = Array(-1, -2, -3)
121-
assert(docVertexIds.forall(i => !LDA.isTermVertex((i.toLong, 0))))
122-
assert(termIds.map(LDA.term2index) === termVertexIds)
123-
assert(termVertexIds.map(i => LDA.index2term(i.toLong)) === termIds)
124-
assert(termVertexIds.forall(i => LDA.isTermVertex((i.toLong, 0))))
121+
assert(docVertexIds.forall(i => !LDA.EMLearningStateInitializer.isTermVertex((i.toLong, 0))))
122+
assert(termIds.map(LDA.EMLearningStateInitializer.term2index) === termVertexIds)
123+
assert(termVertexIds.map(i => LDA.EMLearningStateInitializer.index2term(i.toLong)) === termIds)
124+
assert(termVertexIds.forall(i => LDA.EMLearningStateInitializer.isTermVertex((i.toLong, 0))))
125125
}
126126
}
127127

0 commit comments

Comments
 (0)