type: <fix> message: <require the group variable when needed>

jasonmoy28 · Dec 21, 2023 · e89a443 · e89a443
1 parent 1cae3f3
commit e89a443
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 3 deletions.
diff --git a/R/center_group_mean.R b/R/center_group_mean.R
@@ -21,6 +21,11 @@
 center_group_mean = function(data,cols,group,keep_original=TRUE){
   cols = enquo(cols)
   group = enquo(group)
+
+  group_num = data %>% dplyr::select(!!enquo(group)) %>% names %>% length()
+  if (group_num == 0) {
+   stop('Group variable need to be specified') 
+  }
   original_df = data %>% dplyr::select(!!cols)
   return_df = data %>%
     dplyr::group_by(dplyr::across(!!group)) %>%
@@ -32,3 +37,4 @@ center_group_mean = function(data,cols,group,keep_original=TRUE){
   }
   return(return_df)
 }
+center_group_mean(iris,where(is.numeric),group = Species)
diff --git a/R/center_mlm.R b/R/center_mlm.R
@@ -23,7 +23,12 @@ center_mlm = function(data,cols,group,keep_original = TRUE){
   cols = data %>% dplyr::select(!!enquo(cols)) %>% names()
   group = enquo(group)
   group_name = data %>% dplyr::select(!!enquo(group)) %>% names()
-
+
+  group_num = group_name %>% length()
+  if (group_num == 0) {
+    stop('Group variable need to be specified') 
+  }
+
   # aggregate mean
   mean_data = data %>%
     dplyr::group_by(dplyr::across(!!group)) %>% 

diff --git a/R/z_score_mlm_categorical.R b/R/z_score_mlm_categorical.R
@@ -32,7 +32,10 @@ z_scored_mlm_categorical = function(data,cols,dummy_coded = NA,group,keep_origin
   dummy_coded = data %>% dplyr::select(!!enquo(dummy_coded)) %>% names()
   group = enquo(group)
   group_name = data %>% dplyr::select(!!enquo(group)) %>% names()
-
+  group_num = group_name %>% length()
+  if (group_num == 0) {
+    stop('Group variable need to be specified') 
+  }
 
   # aggregated group mean
   mean_data = data %>%

diff --git a/R/z_scored_group_mean.R b/R/z_scored_group_mean.R
@@ -16,6 +16,10 @@
 z_scored_group_mean = function(data,cols,group,keep_original=TRUE) {
   cols = enquo(cols)
   group = enquo(group)
+  group_num = data %>% dplyr::select(!!enquo(group)) %>% names %>% length()
+  if (group_num == 0) {
+    stop('Group variable need to be specified') 
+  }
   original_df = data %>% dplyr::select(!!cols)
 
   return_df = data %>%

diff --git a/R/z_scored_mlm.R b/R/z_scored_mlm.R
@@ -25,7 +25,10 @@ z_scored_mlm = function(data,cols,group,keep_original = TRUE){
   cols = data %>% dplyr::select(!!enquo(cols)) %>% names()
   group = enquo(group)
   group_name = data %>% dplyr::select(!!enquo(group)) %>% names()
-
+  group_num = group_name %>% length()
+  if (group_num == 0) {
+    stop('Group variable need to be specified') 
+  }
   mean_data = data %>%
     dplyr::group_by(dplyr::across(!!group)) %>% 
     dplyr::summarise(dplyr::across(!!cols,~mean(.,na.rm = TRUE))) %>%