Merge pull request #86 from koheiw/add-test-adjust

Add test for adjust_alpha
koheiw · Sep 4, 2024 · 7287e09 · 7287e09
2 parents f0d18b8 + 18481da
commit 7287e09
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/testthat/test-textmodel_lda.R b/tests/testthat/test-textmodel_lda.R
@@ -121,9 +121,12 @@ test_that("adjust_alpha works", {
 
 	set.seed(1234)
 	lda <- textmodel_lda(dfmt, max_iter = 200, adjust_alpha = 0.5)
+
+	expect_equivalent(rowSums(lda$theta), rep(1.0, ndoc(lda$data)))
+	expect_equivalent(rowSums(lda$phi), rep(1.0, lda$k))
+
 	expect_equal(lda$adjust_alpha, 0.5)
 	expect_true(all(lda$alpha != 0.5))
-	expect_true(all(lda$alpha > 0.25))
 	expect_true(all(lda$epsilon > 0))
 
 	expect_error(