Testing number of specs returned

tests/testthat/test-datamation_sanddance.R

@@ -0,0 +1,27 @@
+test_that("datamation_sanddance returns a frame for the data, one for each group, and two for a summarise step", {
+  library(palmerpenguins)
+
+  pipeline <- "penguins %>% group_by(species)"
+  specs <- datamation_sanddance(pipeline)
+  expect_named(specs, c("data", "group_by"))
+  expect_length(specs[["data"]], 1)
+  expect_length(specs[["group_by"]], 1)
+
+  pipeline <- "penguins %>% group_by(species, island)"
+  specs <- datamation_sanddance(pipeline)
+  expect_named(specs, c("data", "group_by"))
+  expect_length(specs[["data"]], 1)
+  expect_length(specs[["group_by"]], 2)
+
+  pipeline <- "penguins %>% group_by(species, island, sex)"
+  specs <- datamation_sanddance(pipeline)
+  expect_named(specs, c("data", "group_by"))
+  expect_length(specs[["data"]], 1)
+  expect_length(specs[["group_by"]], 3)
+
+  pipeline <- "penguins %>% group_by(species, island) %>% summarize(mean = mean(bill_length_mm))"
+  specs <- datamation_sanddance(pipeline)
+  expect_named(specs, c("data", "group_by"))
+  expect_length(specs[["data"]], 1)
+  expect_length(specs[["group_by"]], 3)
+})