pb210: Lead-210 dating utilities

context("test-utils")

test_that("pb210_core_area() defaults to 1 m^2", {
  expect_equal(pb210_core_area(), 1)
})

test_that("pb210_slice_volume() defaults to 1m^3", {
  expect_equal(pb210_slice_volume(1), 1)
})

test_that("expect_ages_similar works as intended", {
  withr::with_seed(4819, {
    expect_silent(expect_ages_similar(1:10 + runif(10, -1, 1), 1:10, max_delta = 1))
    expect_error(expect_ages_similar(1:10 + runif(10, -1, 1), 1:10, max_delta = 0.1))
  })
})

test_that("expect_ages_similar failes when there are zero values", {
  expect_silent(expect_ages_similar(1:10, 1:10, age_range = 10:12))
  expect_error(expect_ages_similar(1:10, 1:10, age_range = 11:12))
})

test_that("expect_ages_similar failes when there are zero finite values", {
  expect_silent(expect_ages_similar(c(NA_real_, 2:10), 1:10, na.rm = TRUE))
  expect_error(expect_ages_similar(c(NA_real_, 2:10), 1:10, na.rm = FALSE), class = "expectation_failure")
  expect_error(expect_ages_similar(NA_real_, 1:10, na.rm = TRUE), "Zero ages")
})

test_that("with_errors() warns when multiple error types are specified", {
  expect_silent(with_errors(1, 1))
  expect_silent(with_errors(set_errors(1, 1), NA_real_))
  expect_equal(
    with_errors(1, 1),
    with_errors(set_errors(1, 1), NA_real_)
  )
  expect_warning(with_errors(set_errors(1, 1), 1), "Two errors")
})

test_that("extract_errors() warns when multiple error types are specified", {
  expect_silent(extract_errors(1, 1))
  expect_silent(extract_errors(set_errors(1, 1), NA_real_))
  expect_equal(
    extract_errors(1, 1),
    extract_errors(set_errors(1, 1), NA_real_)
  )
  expect_warning(extract_errors(set_errors(1, 1), 1), "Two errors")
})

test_that("trapezoidal integration works", {
  x <- c(1, 2, 3)
  y <- set_errors(c(1, 3, 7), c(1, 1, 1))
  expect_equal(without_errors(integrate_trapezoid(x, y)), c(0, 2, 7))
  expect_equal(without_errors(integrate_trapezoid(x, y, xout = c(1.5, 2.5))), c(1, 4.5))
  expect_equal(extract_errors(integrate_trapezoid(x, y)), c(NA, sqrt(2) / 2, sqrt(4) / 2))
  expect_equal(extract_errors(integrate_trapezoid(x, y, xout = c(1.5, 2.5))), c(NA_real_, NA_real_))
})

test_that("trapezoidal integration works with no errors", {
  expect_identical(
    integrate_trapezoid_no_error(c(1, 2, 3), c(1, 3, 7)),
    without_errors(integrate_trapezoid(c(1, 2, 3), set_errors(c(1, 3, 7), c(1, 1, 1))))
  )
  expect_identical(
    integrate_trapezoid_no_error(c(1, 2, 3), c(1, 3, 7), from = "right"),
    without_errors(
      integrate_trapezoid(c(1, 2, 3), set_errors(c(1, 3, 7), c(1, 1, 1)), from = "right")
    )
  )
})

test_that("input order doesn't matter in trapezoidal integration", {
  x <- c(1, 2, 3)
  y <- c(1, 3, 7)
  expect_equal(
    integrate_trapezoid(x, y),
    integrate_trapezoid(rev(x), rev(y), xout = x)
  )
})

test_that("trapezidal integration works in both directions", {
  x <- c(1, 2, 3)
  y <- c(1, 3, 7)
  expect_equal(
    integrate_trapezoid(x, y, from = "left"),
    integrate_trapezoid(rev(x), y, from = "right")
  )
})

test_that("approx_error() works", {
  x <- 1:10
  y <- seq(10, 100, by = 10)
  err <- seq(0.1, 1, by = 0.1)
  yerr <- set_errors(y, err)

  expect_equal(approx_error(x, yerr, xout = x), yerr)
  expect_equal(sum(is.finite(errors(approx_error(x, yerr, seq(1, 10, by = 0.5))))), 10)
})

test_that("cumulative mass and excess assumptions are checked", {
  mass <- 0:2
  pb210 <- c(10, 1, 0.1)
  expect_silent(check_mass_and_activity(mass, pb210))
  # no non-finite values
  expect_error(check_mass_and_activity(c(0, 1, NA), pb210), "is not TRUE")
  # negative excess values
  expect_error(check_mass_and_activity(mass, c(10, 1, -1)), "is not TRUE")
  # not enough finite values
  expect_error(check_mass_and_activity(mass, c(10, 1, NA)), "is not TRUE")
  expect_error(check_mass_and_activity(mass, c(10, 1, 0)), "is not TRUE")
  # inconsistent lengths
  expect_error(check_mass_and_activity(0:3, pb210), "is not TRUE")
  expect_error(check_mass_and_activity(mass, c(pb210, 0.01)), "is not TRUE")
})