fix for has_theta passed a family object and a test

gavinsimpson · Feb 7, 2022 · 5696b9c · 5696b9c
1 parent 6f16512
commit 5696b9c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/R/distribution-helpers.R b/R/distribution-helpers.R
@@ -95,7 +95,7 @@
 #' p <- theta(m)
 `has_theta` <- function(object) {
     theta_fun <- if (inherits(object, "family")) {
-        family$getTheta
+        object$getTheta
     } else {
         family(object)$getTheta
     }

diff --git a/tests/testthat/test-distribution-helpers.R b/tests/testthat/test-distribution-helpers.R
@@ -40,6 +40,10 @@ test_that("has_theta() returns TRUE for a nb() GAM", {
     expect_true(has_theta(m_nb))
 })
 
+test_that("has_theta() returns TRUE for a nb() GAM", {
+    expect_true(has_theta(nb()))
+})
+
 test_that("has_theta() returns FALSE for a poisson() GAM", {
     expect_false(has_theta(b_pois))
 })