tests/testthat/test-geom-hilbert-outline.R

In ggip: Data Visualization for IP Addresses and Networks

test_that("input validation", {
  address_data <- data.frame(address = ip_address("0.0.0.0"))
  network_data <- data.frame(network = ip_network("0.0.0.0/16"))

  expect_snapshot_error(
    print(ggplot(network_data) + geom_hilbert_outline(aes(ip = network)))
  )
  expect_snapshot_error(
    print(ggplot(address_data) + coord_ip() + geom_hilbert_outline(aes(ip = address)))
  )
  expect_snapshot_error(
    print(ggplot(network_data) + coord_ip(curve = "morton") + geom_hilbert_outline(aes(ip = network)))
  )
})

test_that("alternative ways to specify data/aesthetics", {
  dat <- data.frame(
    ip = ip_network(c("0.0.0.0/2", "128.0.0.0/4"))
  )

  p1 <- ggplot() +
    coord_ip() +
    geom_hilbert_outline(aes(ip = ip), data = dat)

  p2 <- ggplot(dat) +
    coord_ip() +
    geom_hilbert_outline(aes(ip = ip))

  p3 <- ggplot(dat, aes(ip = ip)) +
    coord_ip() +
    geom_hilbert_outline()

  g1 <- layer_grob(p1)[[1]]
  g2 <- layer_grob(p2)[[1]]
  g3 <- layer_grob(p3)[[1]]

  expect_s3_class(g1, "segments")
  expect_s3_class(g2, "segments")
  expect_s3_class(g3, "segments")

  expect_equal(g1$x0, g2$x0)
  expect_equal(g1$x0, g3$x0)
  expect_equal(g1$y0, g2$y0)
  expect_equal(g1$y0, g3$y0)
  expect_equal(g1$x1, g2$x1)
  expect_equal(g1$x1, g3$x1)
  expect_equal(g1$y1, g2$y1)
  expect_equal(g1$y1, g3$y1)
})

test_that("works without data", {
  p <- ggplot() + coord_ip() + geom_hilbert_outline()
  g <- layer_grob(p)[[1]]

  expect_s3_class(g, "segments")
})

test_that("validate drawn segments", {
  expect_segments <- function(curve_order, closed) {
    n_segments <- (2^curve_order + 1)^2
    n_segments <- ifelse(closed, n_segments, n_segments - 2)

    p <- ggplot() +
      coord_ip() +
      geom_hilbert_outline(curve_order = curve_order, closed = closed)

    g <- layer_grob(p)[[1]]

    expect_length(g$x0, n_segments)
  }

  expect_segments(1, FALSE)
  expect_segments(2, TRUE)
  expect_segments(3, FALSE)
  expect_segments(4, TRUE)
})

test_that("networks outside 2D grid raise warning", {
  dat <- data.frame(ip = ip_network("128.0.0.0/4"))

  p <- ggplot(dat, aes(ip = ip)) +
    coord_ip(canvas_network = ip_network("0.0.0.0/2"))

  expect_warning(layer_grob(p + geom_hilbert_outline()))
  expect_silent(layer_grob(p + geom_hilbert_outline(na.rm = TRUE)))
})

test_that("networks without outline are silently ignored", {
  dat <- data.frame(ip = ip_network("128.0.0.0/4"))

  p <- ggplot(dat, aes(ip = ip)) +
    coord_ip() +
    geom_hilbert_outline(curve_order = 2)

  expect_silent(layer_grob(p))
  expect_s3_class(layer_grob(p)[[1]], "zeroGrob")
})