tests/testthat/test_query_model.R

In CausalQueries: Make, Update, and Query Binary Causal Models

context("Query model, query distribution")

testthat::skip_on_cran()
testthat::test_that(

	desc = "Proper error messages.",

	code = {


		model <- make_model("X -> Y") %>% update_model(data.frame(X=0, Y=0), refresh = 0)

		q <- query_distribution(model, query = "Y[X=1] - Y[X=0]", using = "parameters", parameters = 1:6)
		expect_true(q > 0.0555 & q < 0.0556)

		expect_message(query_model(model, query =  "X==1", given = "X==2"))

		expect_error(query_model(model))
		expect_error(query_model(model, query = 1, queries =2))

		q <- query_model(model, query = "Y[X=1] - Y[X=0]", case_level = FALSE, using = "priors")

		expect_true(is.data.frame(q))
    expect_true(q$cred.low < -.5 & q$cred.low > -0.7)
    expect_true(q$cred.high < .7 & q$cred.high > .5)

		q <- query_model(model, query = "Y[X=1] - Y[X=0]", using = "parameters", parameters = list(c(.5, .5, 0, 0, 1, 0)))
		expect_true(is.data.frame(q))

		q <- query_model(model, query = "Y[X=1] - Y[X=0]", using = c("priors", "parameters"), parameters = list(c(.5, .5, 0, 0, 1, 0)))
		expect_true(round(q[2,5]) == 1)

		q <-query_model(
		                model,
		                query = list(ATE = "Y[X=1] - Y[X=0]", Share_positive = "Y[X=1] > Y[X=0]"),
		                using = c("parameters", "priors", "posteriors"),
		                expand_grid = TRUE, stats = NULL)
		expect_true(is.data.frame(q))

    # Use saved type_distribution
		model <- make_model("X -> Y") %>% update_model(data.frame(X=0, Y=0), refresh = 0)

		q <-query_model(
		  model,
		  query = "Y[X=1] - Y[X=0]",
		  using = "posteriors")
		expect_true(is.data.frame(q))
		expect_true(q$cred.low > -0.55 & q$cred.low < -0.45)
		expect_true(q$cred.high > 0.65 & q$cred.high < 0.75)

			}
)