tests/testthat/test-handle_missings.R

In yardstick: Tidy Characterizations of Model Performance

test_that("yardstick_remove_missing works", {
  expect_identical(
    yardstick_remove_missing(1:10, 1:10, 1:10),
    list(
      truth = 1:10,
      estimate = 1:10,
      case_weights = 1:10
    )
  )

  expect_identical(
    yardstick_remove_missing(c(1:4, NA, NA, 7:10), 1:10, 1:10),
    list(
      truth = c(1:4, 7:10),
      estimate = c(1:4, 7:10),
      case_weights = c(1:4, 7:10)
    )
  )

  expect_identical(
    yardstick_remove_missing(1:10, c(1:4, NA, NA, 7:10), 1:10),
    list(
      truth = c(1:4, 7:10),
      estimate = c(1:4, 7:10),
      case_weights = c(1:4, 7:10)
    )
  )

  expect_identical(
    yardstick_remove_missing(1:10, 1:10, c(1:4, NA, NA, 7:10)),
    list(
      truth = c(1:4, 7:10),
      estimate = c(1:4, 7:10),
      case_weights = c(1:4, 7:10)
    )
  )

  expect_identical(
    yardstick_remove_missing(1:10, c(1:4, NA, NA, 7:10), 1:10),
    list(
      truth = c(1:4, 7:10),
      estimate = c(1:4, 7:10),
      case_weights = c(1:4, 7:10)
    )
  )

  expect_identical(
    yardstick_remove_missing(c(NA, 2:10), c(1:9, NA), c(1:4, NA, NA, 7:10)),
    list(
      truth = c(2:4, 7:9),
      estimate = c(2:4, 7:9),
      case_weights = c(2:4, 7:9)
    )
  )
})

test_that("yardstick_any_missing works", {
  expect_false(
    yardstick_any_missing(1:10, 1:10, 1:10)
  )

  expect_true(
    yardstick_any_missing(c(1:4, NA, NA, 7:10), 1:10, 1:10)
  )

  expect_true(
    yardstick_any_missing(1:10, c(1:4, NA, NA, 7:10), 1:10)
  )

  expect_true(
    yardstick_any_missing(1:10, 1:10, c(1:4, NA, NA, 7:10))
  )

  expect_true(
    yardstick_any_missing(1:10, c(1:4, NA, NA, 7:10), 1:10)
  )

  expect_true(
    yardstick_any_missing(c(NA, 2:10), c(1:9, NA), c(1:4, NA, NA, 7:10))
  )
})