From 6c6253f66f3d80885ef548138627b2309fab47ae Mon Sep 17 00:00:00 2001
From: "Mattan S. Ben-Shachar" <mattansb@msbstats.info>
Date: Sat, 7 Dec 2024 20:40:45 +0200
Subject: [PATCH] add tests

---
 tests/testthat/test-convert_between.R | 24 ++++++++++++++++++++++++
 tests/testthat/test-interpret.R       | 23 +++++++++++++++++++++++
 2 files changed, 47 insertions(+)

diff --git a/tests/testthat/test-convert_between.R b/tests/testthat/test-convert_between.R
index 18cce8c1..0acdea3b 100644
--- a/tests/testthat/test-convert_between.R
+++ b/tests/testthat/test-convert_between.R
@@ -18,6 +18,30 @@ test_that("exact OR to d", {
   expect_equal(cor(oddsratio_to_r(OR), d_to_r(d)), 1, tolerance = 0.0002)
   expect_equal(oddsratio_to_r(1), 0, tolerance = 0.0001)
   expect_equal(oddsratio_to_r(OR, p0), d_to_r(d))
+
+
+  # From Chen et al 2010
+  chen_tab_1 <- as.matrix(
+    read.table(
+      text = "p0    OR_1    OR_2    OR_3
+        0.0100  1.6814  3.4739  6.7128
+        0.0200  1.6146  3.1332  5.7486
+        0.0300  1.5733  2.9535  5.2592
+        0.0400  1.5455  2.8306  4.9471
+        0.0500  1.5228  2.7416  4.7233
+        0.0600  1.5060  2.6741  4.5536
+        0.0700  1.4926  2.6177  4.4191
+        0.0800  1.4811  2.5707  4.3097
+        0.0900  1.4709  2.5309  4.2167
+        0.1000  1.4615  2.4972  4.1387",
+      header = TRUE
+    )
+  )
+
+  for (i in seq_len(nrow(chen_tab_1))) {
+    d_recovered <- oddsratio_to_d(chen_tab_1[i, 2:4], p0 = chen_tab_1[i, 1])
+    expect_equal(d_recovered, c(0.2, 0.5, 0.8), tolerance = 0.01, ignore_attr = TRUE)
+  }
 })
 
 
diff --git a/tests/testthat/test-interpret.R b/tests/testthat/test-interpret.R
index 31433fc9..b385b160 100644
--- a/tests/testthat/test-interpret.R
+++ b/tests/testthat/test-interpret.R
@@ -19,6 +19,29 @@ test_that("interpret generic", {
   expect_equal(interpret(c(0, 1), r2)[], c("few", "many"), ignore_attr = TRUE)
 })
 
+
+test_that("interpret matrix / array", {
+  # Matrix
+  r <- cor(mtcars[, 8:11])
+  out <- interpret_r(r)
+
+  expect_equal(dim(out), dim(r))
+  expect_equal(dimnames(out), dimnames(r))
+
+  out2 <- interpret_r(as.vector(r))
+  expect_equal(out2, as.vector(out), ignore_attr = TRUE)
+
+  # Array
+  set.seed(111)
+  p <- array(runif(3 * 2 * 4, max = 0.1), dim = c(3, 2, 4))
+  out <- interpret_p(p, rules = "rss")
+
+  expect_equal(dim(out), dim(p))
+
+  out2 <- interpret_p(as.vector(p), rules = "rss")
+  expect_equal(out2, as.vector(out), ignore_attr = TRUE)
+})
+
 # interpret types ----
 test_that("interpret_r", {
   expect_equal(interpret_r(0.21)[1], "medium")