tests/testthat/test-compute_deriv_nn.R

In boyuren158/GP-CERF: Gaussian Processes for Estimating Causal Exposure Response Curves

test_that("compute_deriv_nn works as expected!", {

  set.seed(765)
  data <- generate_synthetic_data(sample_size = 200)
  gps_m <- estimate_gps(cov_mt = data[, -(1:2)],
                        w_all = data$treat,
                        sl_lib = c("SL.xgboost"),
                        dnorm_log = FALSE)

  wi <- 4.8

  deriv_val <- compute_deriv_nn(w = wi,
                                w_obs = data$treat,
                                gps_m = gps_m,
                                y_obs = data$Y,
                                hyperparam = c(0.1, 0.2, 1),
                                n_neighbor = 20,
                                block_size = 1000)

  expect_equal(length(deriv_val), 1L)
  expect_equal(deriv_val[1, 1], 1.558466, tolerance = 0.00001)
  expect_true(is.matrix(deriv_val))
})