tests/testthat/test-plot_plotly.R

In shosaco/vistime: Pretty Timelines in R

# test ranges
dat <- data.frame(start = c("2019-01-01 00:00", "2019-01-01 04:00"),
                  end = c("2019-01-01 05:00", "2019-01-05 06:00"),
                  event = 1:2)

test_that("class is htmlwidget", {
    expect_s3_class(vistime(dat), "htmlwidget")
})

test_that("background_lines", {
    for (bg in c(1, 5, 10)) {
        expect_equal(bg + 2,
                     length(
                         suppressWarnings(plotly::plotly_build(
                             vistime(dat, background_lines = bg)
                         )$x$layout$shapes)
                     ))
    }

    expect_true(suppressWarnings(plotly::plotly_build(vistime(
        dat, background_lines = NULL
    ))$x$layout$xaxis$showgrid))
    expect_false(suppressWarnings(plotly::plotly_build(vistime(
        dat, background_lines = 10
    ))$x$layout$xaxis$showgrid))
})


test_that("color is same as in df", {
  dat$color <- "green"
  res <- c()
  for(x in vistime(dat)$x$attrs) if(x$mode == "lines" && length(x$x) == 2) res <- c(res, x$line$color)
  expect_setequal(res, dat$color)

  dat$color <- c("yellow", "blue")
  res <- c()
  for(x in vistime(dat)$x$attrs) if(x$mode == "lines" && length(x$x) == 2) res <- c(res, x$line$color)
  expect_setequal(res, c("yellow", "blue"))
})

test_that("start and end", {
  res <- c()
  for(x in vistime(dat)$x$attrs) if(x$mode == "lines" && length(x$y) == 1) res <- c(res, x$x)

  expect_setequal(res,sapply(unlist(lapply(dat[c("start", "end")], as.character)), as.POSIXct))
})


# test events
dat <- data.frame(
  event = 1:2, start = as.Date(c("2019-01-01", "2019-01-10")),
  col = "green", fontcol = "black"
)

test_that("data having no real events (only ranges)", {
  dat$end <- dat$start + 5
  res <- list()
  for(x in vistime(dat)$x$attrs) if(x$mode == "markers") res <- append(res, x)
  expect_equal(res, list())
})

dat$end = NA
test_that("class is htmlwidget", {
    expect_s3_class(vistime(dat), "htmlwidget")
})

relevant_dat <- vistime(dat)$x$attrs

test_that("Number of markers", {
  num_markers <- 0
  for(x in relevant_dat) if(x$mode == "markers") num_markers <- num_markers + length(x$x)
  expect_equal(num_markers, nrow(dat))
})

test_that("Symbol is circle", {
  res <- c()
  for(x in relevant_dat) if(x$mode == "markers") res <- c(res, x$marker$symbol)
  expect_equal(unique(res), "circle")
})

test_that("x Values", {
  res <- c()
  for(x in relevant_dat) if(x$mode == "markers") res <- c(res, as.integer(x$x))
  expect_equal(res, as.integer(as.POSIXct(dat$start)))
})