# Small data set for testing purposes
aus_mortality <- aus_mortality  |>
  filter(Year <= 1903,
  Sex == "female",
  State %in% c("Victoria", "Tasmania"),
  Age > 80)

test_that("classes", {
  expect_s3_class(aus_mortality |> filter(State == "Victoria"), "vital")
  expect_s3_class(aus_mortality |> select(Exposure), "vital")
  expect_s3_class(aus_mortality |> slice(10), "vital")
  expect_s3_class(aus_mortality |> arrange(Age, Sex, State, Code, Year), "vital")
  expect_s3_class(aus_mortality |> mutate(mx = Exposure/Deaths), "vital")
  expect_s3_class(aus_mortality |> head(10), "vital")
  expect_s3_class(aus_mortality |> tail(100), "vital")
  expect_s3_class(bind_rows(aus_mortality |> filter(Year == 1901),
      aus_mortality |> filter(Year == 1902)), "vital")
  expect_s3_class(bind_cols(aus_mortality |> select(-Exposure),
      Exposure = aus_mortality$Exposure), "vital")
  expect_s3_class(aus_mortality |> transmute(mx = Exposure/Deaths), "vital")
  expect_s3_class(aus_mortality |> relocate(Year, Exposure, Age), "vital")
  expect_s3_class(aus_mortality |> summarise(Deaths = mean(Deaths)), "vital")
  expect_s3_class(aus_mortality |> group_by(Sex) |> summarise(Deaths = mean(Deaths)), "vital")
  expect_s3_class(aus_mortality |> group_by(Sex) |> ungroup(), "vital")
  expect_s3_class(left_join(aus_mortality |> select(Exposure),  aus_mortality |> select(Deaths)), "vital")
})