Merge pull request #28 from Kyoshido/Kyoshido-patch-1

Update simStructure.R
statistikat · Aug 1, 2023 · 4b7bd5b · 4b7bd5b
2 parents 5871417 + 4e21719
commit 4b7bd5b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/R/simStructure.R b/R/simStructure.R
@@ -94,6 +94,11 @@ simStructure <- function(dataS, method=c("direct", "multinom", "distribution"),
     strata <- factor(rep(1, nrow(dataS@data)))
   }
 
+  # Check levels for multinom method
+  if ( method == "multinom" & length(levels(strata)) < 2 ) {
+    stop("Your strata region must have 2 or more levels!\n")
+  }
+
   ##### set up household structure
 
   # generate variables on household level (indicated by H)