Skip to content

Commit 37e4e6a

Browse files
committed
add a test
1 parent faabe69 commit 37e4e6a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -505,6 +505,8 @@ class LDASuite extends SparkFunSuite with MLlibTestSparkContext {
505505
assert(distributedModel.topicConcentration === sameDistributedModel.topicConcentration)
506506
assert(distributedModel.gammaShape === sameDistributedModel.gammaShape)
507507
assert(distributedModel.globalTopicTotals === sameDistributedModel.globalTopicTotals)
508+
assert(distributedModel.logLikelihood ~== sameDistributedModel.logLikelihood absTol 1e-6)
509+
assert(distributedModel.logPrior ~== sameDistributedModel.logPrior absTol 1e-6)
508510

509511
val graph = distributedModel.graph
510512
val sameGraph = sameDistributedModel.graph

0 commit comments

Comments
 (0)