Add check for invalid probs weights args

smishr · Oct 31, 2022 · ace6bbe · ace6bbe
1 parent c14c088
commit ace6bbe
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/SurveyDesign.jl b/src/SurveyDesign.jl
@@ -32,7 +32,11 @@ struct SimpleRandomSample <: AbstractSurveyDesign
         if isa(probs, Symbol)
             probs = data[!, probs]
         end
-
+        if !( sum(probs) ≈ 1 )
+            error("Sampling probabilities for SRS must sum to 1")
+        elseif !(sum(1 ./ weights) ≈ 1)
+            error("Inverse of weights should for SRS must sum to 1")
+        end
         if ignorefpc
             @warn "assuming all weights are equal to 1.0"
             weights = ones(nrow(data))