library(testthat)

test_that("transform_data returns a data frame", {
  df <- transform_data(data.frame(
    age=c(1, 2, 3),
    status=c(1, 0, 1)
  ))
  expect_type(df, "list")
})

test_that("transform_data returns the correct number of rows", {
  df <- transform_data(data.frame(
    age=c(1, 2, 3),
    status=c(1, 0, 1)
  ))
  expect_equal(nrow(df), 3)
})

test_that("transform_data returns the correct values", {
  df <- transform_data(
    data.frame(
      age=c(1, 1, 2, 2, 3),
      status=c(1, 0, 1, 1, 0)
    )
  )
  expect_equal(df$pos, c(1, 2, 0))
  expect_equal(df$tot, c(2, 2, 1))
})

test_that("est_foi returns a numeric vector", {
  t <- c(1, 2, 3, 4)
  sp <- c(0.1, 0.2, 0.3, 0.4)
  expect_type(est_foi(t, sp), "double")
})

test_that("est_foi returns the correct output", {
  t <- c(1, 2, 3, 4)
  sp <- c(0.1, 0.2, 0.3, 0.4)
  expect_equal(est_foi(t, sp), c(0.125, 0.14285714))
})