test for absence/presence of guide differently

wilkelab · Jan 17, 2024 · e43884a · e43884a
1 parent 54c8670
commit e43884a
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/tests/testthat/test_scale_cyclical.R b/tests/testthat/test_scale_cyclical.R
@@ -12,7 +12,11 @@ test_that("basic tests", {
   expect_equal(d$colour, rep(c("#F00000", "#0000F0"), 13))
 
   # make sure there is no legend being generated
-  expect_equal("guide-box" %in% ggplotGrob(p)$layout$name, FALSE)
+  if ("get_guide_data" %in% getNamespaceExports("ggplot2")) {
+    expect_null(get_guide_data(p, "colour"))
+  } else {
+    expect_equal("guide-box" %in% ggplotGrob(p)$layout$name, FALSE)
+  }
 
   # once again, different aesthetic, different cyclical pattern, now with legend
   p <- ggplot(df, aes(x, y, label=letters, color=factor(x))) + geom_text() +
@@ -23,7 +27,11 @@ test_that("basic tests", {
   expect_equal(d$colour[order(d$x)], rep(c("#F00000", "#0000F0", "#F0F000"), 9)[1:26])
 
   # make sure there is a legend
-  expect_equal("guide-box" %in% ggplotGrob(p)$layout$name, TRUE)
+  if ("get_guide_data" %in% getNamespaceExports("ggplot2")) {
+    expect_s3_class(get_guide_data(p, "colour"), "data.frame")
+  } else {
+    expect_equal("guide-box" %in% ggplotGrob(p)$layout$name, TRUE)
+  }
 
   # test that breaks must match labels
   expect_error(