tests/testthat/test-annotations.R

In angusmoore/arphit: RBA-style R Plots

## Labels ===================

# Labels
test_that("Labels", {
  foo <- arphitgg() +
    agg_label("Text", x = 2002, y = 0.2, panel = "1", colour = "red")
  bar <- arphitgg() +
    agg_label("Text", x = 2002, y = 0.2, panel = "1", colour = "red") +
    agg_label("Second label",
              x = 2003,
              y = -0.2,
              panel = "1",
              colour = "green",
              size = 40)
  expect_true(check_graph(foo, "annotations-label1"))
  expect_true(check_graph(bar, "annotations-label2"))
})

## Arrows ==================

test_that("Arrows", {
  foo <- arphitgg() +
    agg_arrow(tail.x = 2000,
              tail.y = 0,
              head.x = 2001,
              head.y = 2,
              colour = "red",
              panel = "1")
  bar <- arphitgg() +
    agg_arrow(tail.x = 2000,
              tail.y = 0,
              head.x = 2001,
              head.y = 2,
              colour = "red",
              panel = "1") +
    agg_arrow(tail.x = 2001,
              tail.y = 0,
              head.x = 2002,
              head.y = 2,
              colour = "green",
              panel = "1",
              lwd = 2)
  expect_true(check_graph(foo, "annotations-arrows1"))
  expect_true(check_graph(bar, "annotations-arrows2"))
})

## Background shading ================

test_that("Background shading", {
  foo <- arphitgg() + agg_bgshading(x1 = 2001, x2 = 2002, panel = "1")
  bar <- arphitgg() + agg_bgshading(y1 = 0.5, y2 = -0.5, panel = "1")
  baz <- arphitgg() +
    agg_bgshading(x1 = 2001,
                  x2 = 2002,
                  y1 = 0.5,
                  y2 = -0.5,
                  colour = "red",
                  panel = "1")

  expect_true(check_graph(foo, "annotations-bgshading1"))
  expect_true(check_graph(bar, "annotations-bgshading2"))
  expect_true(check_graph(baz, "annotations-bgshading3"))
})

## AB lines ================

test_that("AB lines", {
  foo <- arphitgg() + agg_vline(x = 2001, colour = RBA["Blue1"], panel = "1")
  bar <- arphitgg() + agg_hline(y = -0.5, colour = RBA["Red1"], panel = "1") +
    agg_vline(x = 2001, colour = RBA["Blue1"], panel = "1")
  baz <- arphitgg() +
    agg_abline(x1 = 2000, y1 = -0.1, x2 = 2002, y2 = 0.5, panel = "1")
  p <- arphitgg() + agg_vline(x = 2002, lty = 5, lwd = 20, panel = "1")

  expect_true(check_graph(foo, "annotations-ablines1"))
  expect_true(check_graph(bar, "annotations-ablines2"))
  expect_true(check_graph(baz, "annotations-ablines3"))
  expect_true(check_graph(p, "annotations-set-lty-lwd"))
})

## Extrapolating on missing arguments for ab lines ============

test_that("Extrapolate ab lines arguments", {
  expect_error(agg_vline(x = 2001),
               "argument \"panel\" is missing, with no default")
  expect_error(
    agg_abline(
      x1 = 2000,
      y1 = -1,
      y2 = 0,
      panel = "1"
    ),
    "Line is missing",
    fixed = TRUE
  )
  expect_error(
    agg_abline(
      x1 = 2000,
      x2 = 2001,
      y1 = -1,
      panel = "1"
    ),
    "Line is missing",
    fixed = TRUE
  )
  expect_error(
    agg_abline(
      x2 = 2000,
      y1 = -1,
      y2 = 0,
      panel = "1"
    ),
    "Line is missing",
    fixed = TRUE
  )
  expect_error(
    agg_abline(
      x1 = 2000,
      x2 = 2001,
      y2 = -1,
      panel = "1"
    ),
    "Line is missing",
    fixed = TRUE
  )
})