Fix reference_subgroup check

WHOequity · WHOequity · commit 3e7e0406a7ed · 2024-11-22T20:28:12.000+01:00
diff --git a/R/mdru.R b/R/mdru.R
@@ -99,8 +99,8 @@ mdru <- function(est,
     if (!is.numeric(se))
       stop('Standard errors need to be numeric')
   }
-  if (sum(reference_subgroup) !=1 ) {
-    stop('Reference subgroup variable is missing')
+  if (sum(reference_subgroup) < 1 ) {
+    stop('Reference subgroup is missing')
   }
   if (sum(reference_subgroup) > 1) {
     stop('Reference subgroup variable must identify one reference subgroup
diff --git a/R/mdrw.R b/R/mdrw.R
@@ -113,8 +113,8 @@ mdrw <- function(est,
   if (all(pop == 0)) {
     stop('Population variable is of size 0 in all subgroups')
   }
-  if (sum(reference_subgroup) !=1 ) {
-    stop('Reference subgroup variable is missing')
+  if (sum(reference_subgroup) < 1 ) {
+    stop('Reference subgroup is missing')
   }
   if (sum(reference_subgroup) > 1) {
     stop('Reference subgroup variable must identify one reference subgroup

Original file line number	Diff line number	Diff line change
`@@ -99,8 +99,8 @@ mdru <- function(est,`
`99`	`99`	`if (!is.numeric(se))`
`100`	`100`	`stop('Standard errors need to be numeric')`
`101`	`101`	`}`
`102`		`- if (sum(reference_subgroup) !=1 ) {`
`103`		`- stop('Reference subgroup variable is missing')`
	`102`	`+ if (sum(reference_subgroup) < 1 ) {`
	`103`	`+ stop('Reference subgroup is missing')`
`104`	`104`	`}`
`105`	`105`	`if (sum(reference_subgroup) > 1) {`
`106`	`106`	`stop('Reference subgroup variable must identify one reference subgroup`
Original file line number	Diff line number	Diff line change
`@@ -113,8 +113,8 @@ mdrw <- function(est,`
`113`	`113`	`if (all(pop == 0)) {`
`114`	`114`	`stop('Population variable is of size 0 in all subgroups')`
`115`	`115`	`}`
`116`		`- if (sum(reference_subgroup) !=1 ) {`
`117`		`- stop('Reference subgroup variable is missing')`
	`116`	`+ if (sum(reference_subgroup) < 1 ) {`
	`117`	`+ stop('Reference subgroup is missing')`
`118`	`118`	`}`
`119`	`119`	`if (sum(reference_subgroup) > 1) {`
`120`	`120`	`stop('Reference subgroup variable must identify one reference subgroup`