Skip to content

Commit

Permalink
follow lintr suggestions
Browse files Browse the repository at this point in the history
  • Loading branch information
wibeasley committed Aug 12, 2022
1 parent f6e7e7c commit 18a1c5b
Show file tree
Hide file tree
Showing 3 changed files with 42 additions and 22 deletions.
40 changes: 24 additions & 16 deletions R/table-nih-enrollment.R
Original file line number Diff line number Diff line change
Expand Up @@ -118,8 +118,12 @@
#' @export
table_nih_enrollment <- function(
d,
d_lu_gender=NULL, d_lu_race=NULL, d_lu_ethnicity=NULL,
variable_gender="gender", variable_race="race", variable_ethnicity="ethnicity"
d_lu_gender = NULL,
d_lu_race = NULL,
d_lu_ethnicity = NULL,
variable_gender = "gender",
variable_race = "race",
variable_ethnicity = "ethnicity"
) {
checkmate::assert_data_frame(d , any.missing = FALSE)
checkmate::assert_data_frame(d_lu_gender , any.missing = FALSE, null.ok = TRUE)
Expand Down Expand Up @@ -162,34 +166,34 @@ table_nih_enrollment <- function(
race = !!variable_race ,
ethnicity = !!variable_ethnicity
)
if( !is.null(d_lu_gender) ) {
if (!is.null(d_lu_gender)) {
d <- d |>
dplyr::left_join(d_lu_gender, by=c("gender" = "input")) |>
dplyr::left_join(d_lu_gender, by = c("gender" = "input")) |>
dplyr::select(-.data$gender) |>
dplyr::rename(gender = .data$displayed)
}

if( !is.null(d_lu_race) ) {
if (!is.null(d_lu_race)) {
d <- d |>
dplyr::left_join(d_lu_race, by=c("race" = "input")) |>
dplyr::left_join(d_lu_race, by = c("race" = "input")) |>
dplyr::select(-.data$race) |>
dplyr::rename(race = .data$displayed)
}

if( !is.null(d_lu_ethnicity) ) {
if (!is.null(d_lu_ethnicity)) {
d <- d |>
dplyr::left_join(d_lu_ethnicity, by=c("ethnicity" = "input")) |>
dplyr::left_join(d_lu_ethnicity, by = c("ethnicity" = "input")) |>
dplyr::select(-.data$ethnicity) |>
dplyr::rename(ethnicity = .data$displayed)
}

d_count <- d |>
d |>
dplyr::count(.data$gender, .data$race, .data$ethnicity) |>
dplyr::full_join(d_possible, by = c("gender", "race", "ethnicity")) |>
dplyr::mutate(
gender = factor(.data$gender , levels=levels_gender ),
race = factor(.data$race , levels=levels_race ),
ethnicity = factor(.data$ethnicity, levels=levels_ethnicity ),
gender = factor(.data$gender , levels = levels_gender ),
race = factor(.data$race , levels = levels_race ),
ethnicity = factor(.data$ethnicity, levels = levels_ethnicity ),
n = dplyr::coalesce(.data$n, 0L)
) |>
dplyr::select(.data$gender, .data$race, .data$ethnicity, .data$n) |>
Expand All @@ -199,8 +203,12 @@ table_nih_enrollment <- function(
#' @export
table_nih_enrollment_pretty <- function(
d,
d_lu_gender=NULL, d_lu_race=NULL, d_lu_ethnicity=NULL,
variable_gender="gender", variable_race="race", variable_ethnicity="ethnicity"
d_lu_gender = NULL,
d_lu_race = NULL,
d_lu_ethnicity = NULL,
variable_gender = "gender",
variable_race = "race",
variable_ethnicity = "ethnicity"
) {
column_order <- c(
"race",
Expand All @@ -223,11 +231,11 @@ table_nih_enrollment_pretty <- function(
gender_ethnicity = paste0(.data$gender, " by ", .data$ethnicity)
) |>
dplyr::select(-.data$gender, -.data$ethnicity) |>
tidyr::spread(key=.data$gender_ethnicity, value=.data$n) |>
tidyr::spread(key = .data$gender_ethnicity, value = .data$n) |>
dplyr::select(!!column_order) |>
knitr::kable(
format = "html",
format.args = list(big.mark=","),
format.args = list(big.mark = ","),
escape = FALSE,
col.names = c(
"Racial\nCategories",
Expand Down
20 changes: 16 additions & 4 deletions tests/testthat/test-table-nih-enrollment.R
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ test_that("ds_1b --ethnicity metadata required", {
"Unknown" , "Unknown/Not Reported Ethnicity"
)

observed <- table_nih_enrollment(d_1b, d_lu_ethnicity=d_lu_ethnicity)
observed <- table_nih_enrollment(d_1b, d_lu_ethnicity = d_lu_ethnicity)
expect_equal(observed, expected_1)
})

Expand Down Expand Up @@ -112,7 +112,13 @@ test_that("ds_1c --all metadata required", {
"Unknown" , "Unknown/Not Reported Ethnicity"
)

observed <- table_nih_enrollment(d_1c, d_lu_gender=d_lu_gender, d_lu_race=d_lu_race, d_lu_ethnicity=d_lu_ethnicity)
observed <-
table_nih_enrollment(
d_1c,
d_lu_gender = d_lu_gender,
d_lu_race = d_lu_race,
d_lu_ethnicity = d_lu_ethnicity
)
expect_equal(observed, expected_1)
})

Expand All @@ -121,10 +127,16 @@ test_that("ds_1d --different variable names", {
dplyr::rename(
vg = gender,
vr = race,
ve = ethnicity
ve = ethnicity,
)

observed <- table_nih_enrollment(ds_1d, variable_gender="vg", variable_race="vr", variable_ethnicity="ve")
observed <-
table_nih_enrollment(
ds_1d,
variable_gender = "vg",
variable_race = "vr",
variable_ethnicity = "ve"
)
expect_equal(observed, expected_1)
})

Expand Down
4 changes: 2 additions & 2 deletions vignettes/nih-enrollment-html.Rmd
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ Local Data Source
### Establish Datasets

```{r local-establish}
path <- system.file("misc/example-data-1.csv", package="codified")
path <- system.file("misc/example-data-1.csv", package = "codified")
col_types <- readr::cols_only(
record_id = readr::col_integer(),
name_last = readr::col_character(),
Expand All @@ -50,7 +50,7 @@ col_types <- readr::cols_only(
race = readr::col_integer(),
ethnicity = readr::col_integer()
)
ds <- readr::read_csv(path, col_types=col_types) |>
ds <- readr::read_csv(path, col_types = col_types) |>
dplyr::mutate(
gender = as.character(gender),
race = as.character(race),
Expand Down

0 comments on commit 18a1c5b

Please sign in to comment.