tests/testthat/helper-data.R

In highdir: Unified Data Visualization Framework for Dynamic and Static Graphics

#' Standard health-survey data frame used across tests
survey_df <- data.frame(
  age = rep(c("18-24", "25-34", "35-44", "45-54"), each = 2),
  sex = rep(c("Male", "Female"), 4),
  pct = c(42, 38, 55, 61, 48, 52, 60, 57),
  lo  = c(37, 33, 50, 56, 43, 47, 55, 52),
  hi  = c(47, 43, 60, 66, 53, 57, 65, 62),
  n   = c(120, 115, 200, 210, 180, 175, 160, 155)
)

#' Standard pie data frame
pie_df <- data.frame(
  category = c("Aldri", "Sjelden", "Av og til", "Ofte"),
  value    = c(35, 25, 25, 15)
)

#' Use in colors and palettes testing
df2 <- data.frame(
  x   = rep(c("A","B","C"), 2),
  y   = c(1,2,3,4,5,6),
  grp = rep(c("Male","Female"), each = 3)
)

spec2 <- hd_spec(df2, "x", "y", group = "grp")

# Helper used in tests for colors and paletts for ggplot2
.extract_gg_fill_values <- function(fig) {
  idx <- which(vapply(fig$scales$scales,
                      function(s) "fill" %in% s$aesthetics,
                      logical(1)))
  if (!length(idx)) return(NULL)
  fig$scales$scales[[idx[1]]]$palette(
    length(fig$scales$scales[[idx[1]]]$range$range)
  )
}

# For stacked_column
olympics <- data.frame(
  Country   = c("Norway", "Norway", "Norway",
                "Germany", "Germany", "Germany",
                "United States", "United States", "United States",
                "Canada", "Canada", "Canada"),
  Continent = c("Europe", "Europe", "Europe",
                "Europe", "Europe", "Europe",
                "North America", "North America", "North America",
                "North America", "North America", "North America"),
  Medal     = rep(c("Gold", "Silver", "Bronze"), times = 4),
  Count     = c(148, 133, 124,
                102, 98, 65,
                113, 122, 95,
                77, 72, 80),
  pros      = c(
    36.5, 32.8, 30.6,
    38.5, 37.0, 24.5,
    34.2, 37.0, 28.8,
    33.6, 31.4, 34.9
  )
)


twostack_df <- data.frame(
  x = c("2024", "2024", "2024", "2025", "2025", "2025"),
  y = c(10, 20, 15, 12, 18, 22),
  grp = c("g1", "g2", "g3", "g4", "g5", "g6"),
  stack = c("A", "A", "A", "B", "B", "B"),
  stringsAsFactors = FALSE
)