tests/testthat/test-estimate_win_prob_vs_baseline.R

In rangi513/grizbayr: Bayesian Inference for A|B and Bandit Marketing Tests

context("Estimate Win Probability vs Baseline")

test_that("estimate_win_prob_vs_baseline returns only 2 options", {
  input_df <- tibble::tibble(
    option_name = c("A", "B", "C"),
    sum_clicks = c(1000, 1000, 1000),
    sum_conversions = c(100, 120, 110)
  )
  all_option_names <- unique(input_df$option_name)
  output <- estimate_win_prob_vs_baseline(input_df = input_df,
                                          distribution = "conversion_rate",
                                          wrt_option = "A")
  # Subset exists
  expect_true(all(output$option_name %in% all_option_names))
  expect_true(nrow(output) == 2)
})


test_that("estimate_win_prob_vs_baseline_given_posterior handles when wrt_option is the best", {
  # Can't be better than yourself
  input_df <- tibble::tibble(
    option_name = c("A", "B", "C"),
    sum_clicks = c(1000, 1000, 1000),
    sum_conversions = c(100, 120, 110)
  )
  all_option_names <- unique(input_df$option_name)
  output <- estimate_win_prob_vs_baseline(input_df = input_df,
                                          distribution = "conversion_rate",
                                          wrt_option = "B")
  expect_true(all(output$option_name %in% all_option_names))
  expect_true(nrow(output) == 1)
})