Add full collocation analysis (client side only implementation)
Resolves #2
Change-Id: Ib01d89a72b44ff06816b21532b7ea709a4e837b0
diff --git a/tests/testthat/test-collocations.R b/tests/testthat/test-collocations.R
new file mode 100644
index 0000000..0c17144
--- /dev/null
+++ b/tests/testthat/test-collocations.R
@@ -0,0 +1,23 @@
+test_that("collocationScoreQuery works", {
+ kco <- new("KorAPConnection", cache = TRUE, verbose = TRUE)
+ df <- collocationScoreQuery(kco, "Ameisenplage", "heimgesucht", leftContextSize=0, rightContextSize=1)
+ expect_gt(df$logDice, 1)
+ expect_equal(df$ll, ll(df$O1, df$O2, df$O, df$N, df$E, df$w))
+ expect_equal(df$pmi, pmi(df$O1, df$O2, df$O, df$N, df$E, df$w))
+ expect_equal(df$mi2, mi2(df$O1, df$O2, df$O, df$N, df$E, df$w))
+ expect_equal(df$mi3, mi3(df$O1, df$O2, df$O, df$N, df$E, df$w))
+ expect_equal(df$logDice, logDice(df$O1, df$O2, df$O, df$N, df$E, df$w))
+})
+
+test_that("collocationAnalysis works", {
+ kco <- new("KorAPConnection", cache = TRUE, verbose = TRUE)
+ df <- collocationAnalysis(kco, "Ameisenplage", leftContextSize=0, rightContextSize=1, topCollocatesLimit=1, exactFrequencies=FALSE)
+ expect_gt(df$O, df$E)
+ expect_gt(df$logDice, 1)
+})
+
+test_that("collocationAnalysis on unaccounted strings does not error out", {
+ kco <- new("KorAPConnection", cache = TRUE, verbose = TRUE)
+ df <- collocationAnalysis(kco, "XXXXXXXXAmeisenplage")
+ testthat::expect_equal(nrow(df), 0)
+})