tests/testthat/test-Point.R

In cdcepi/predx: Tools for working with predictions

context("Point")

test_that("Creates Point object", {
  expect_is(Point(0.9), "Point")
  expect_is(Point(-1000), "Point")
})

test_that("Point rejects invalid point predictions", {
  expect_error(Point("9"))
  expect_error(Point(NA))
  expect_error(Point(c(15, 25)))
})

test_that("Point data frame objects convert to predx", {
  expect_silent(to_predx(list(data.frame(point=10),
    data.frame(point=15)), rep('Point', 2)))
})

test_that("Generics function", {
  this_pred <- Point(5)
  expect_equal(predx_to_json(this_pred), list(point = 5))
  expect_equal(as.data.frame(this_pred), data.frame(point = 5))
})

test_that("CSV import/export works", {
  fcast <- dplyr::tibble(target = 'x', predx_class = 'Point',
    predx = list(Point(10)))
  csv_file <- tempfile()
  export_csv(fcast, csv_file)
  fcast_import <- import_csv(csv_file)
  expect_equal(as.data.frame(fcast_import), as.data.frame(fcast))
})

test_that("JSON import/export works", {
  fcast <- dplyr::tibble(target = 'x', predx_class = 'Point',
    predx = list(Point(10)))
  json_file <- tempfile()
  export_json(fcast, json_file)
  fcast_import <- import_json(json_file)
  expect_equal(as.data.frame(fcast_import), as.data.frame(fcast))
})