###############################################################################
# Suppressing some functions messages because they only output the information
# on how much time they took.
###############################################################################

test_df <- suppressMessages(dummy_data(10))

###############################################################################
# Keep
###############################################################################

test_that("Different way of passing variables in keep", {
    keep_df1 <- test_df |> keep(year)
    keep_df2 <- test_df |> keep("year")

    expect_identical(keep_df1, keep_df2)
})


test_that("Keep only one variable", {
    keep_df <- test_df |> keep(year)

    expect_equal(ncol(keep_df), 1)

    expect_true("year" %in% names(keep_df))

    expect_identical(keep_df[["year"]], test_df[["year"]])
})


test_that("Keep more than one variable", {
    keep_df <- test_df |> keep(year, age, sex)

    expect_equal(ncol(keep_df), 3)

    expect_true(all(c("year", "sex", "age") %in% names(keep_df)))

    expect_identical(keep_df[["year"]], test_df[["year"]])
    expect_identical(keep_df[["age"]], test_df[["age"]])
    expect_identical(keep_df[["sex"]], test_df[["sex"]])
})


test_that("Keep range of variables", {
    keep_df <- test_df |> keep(state:age)

    expect_equal(ncol(keep_df), 3)

    expect_true(all(c("state", "sex", "age") %in% names(keep_df)))

    expect_identical(keep_df[["state"]], test_df[["state"]])
    expect_identical(keep_df[["age"]], test_df[["age"]])
    expect_identical(keep_df[["sex"]], test_df[["sex"]])
})


test_that("Keep variables starting with letter", {
    keep_df <- test_df |> keep("s:")

    expect_equal(ncol(keep_df), 2)

    expect_true(all(c("state", "sex") %in% names(keep_df)))
})


test_that("Keep variables ending with letter", {
    keep_df <- test_df |> keep(":id")

    expect_equal(ncol(keep_df), 2)

    expect_true(all(c("household_id", "person_id") %in% names(keep_df)))
})


test_that("Keep variables containing letter", {
    keep_df <- test_df |> keep(":on:")

    expect_equal(ncol(keep_df), 3)

    expect_true(all(c("person_id", "first_person", "education") %in% names(keep_df)))
})


test_that("Variables to keep contain a variable name that isn't part of the data frame", {
    expect_message(keep_df1 <- test_df |> keep(year, age, sex, cats, dogs), " ! WARNING: The provided variable to keep")
    keep_df2 <- test_df |> keep(year, age, sex)

    expect_equal(ncol(keep_df1), 3)
    expect_equal(ncol(keep_df2), 3)

    expect_identical(keep_df1, keep_df2)

    expect_true(all(c("year", "sex", "age") %in% names(keep_df1)))
    expect_true(!all(c("cats", "dogs") %in% names(keep_df1)))
})


test_that("Keep only variables that are not part of the data frame", {
    expect_message(keep_df <- test_df |> keep(cats, dogs), " ! WARNING: The provided variable to keep")

    expect_true(nrow(keep_df) == 0)
    expect_true(ncol(keep_df) == 0)

    expect_true(!all(c("cats", "dogs") %in% names(keep_df)))
})


test_that("Keep without any variables provided", {
    keep_df <- test_df |> keep()

    expect_identical(keep_df, test_df)
})


test_that("Keep with sorted variables", {
    unsorted <- test_df |> keep(weight, age)
    sorted   <- test_df |> keep(weight, age, order_vars = TRUE)

    expect_equal(names(unsorted)[1], "age")
    expect_equal(names(sorted)[1], "weight")
})

###############################################################################
# Drop
###############################################################################

test_that("Different way of passing variables in drop", {
    drop_df1 <- test_df |> dropp(year)
    drop_df2 <- test_df |> dropp("year")

    expect_identical(drop_df1, drop_df2)
})


test_that("Drop only one variable", {
    drop_df <- test_df |> dropp(year)

    expect_equal(ncol(drop_df), ncol(test_df) - 1)

    expect_true(!"year" %in% names(drop_df))
})


test_that("Drop more than one variable", {
    drop_df <- test_df |> dropp(year, age, sex)

    expect_equal(ncol(drop_df), ncol(test_df) - 3)

    expect_true(!all(c("year", "sex", "age") %in% names(drop_df)))
})


test_that("Drop range of variables", {
    drop_df <- test_df |> dropp(state:age)

    expect_equal(ncol(drop_df), ncol(test_df) - 3)

    expect_true(!all(c("state", "sex", "age") %in% names(drop_df)))
})


test_that("Drop variables starting with letter", {
    drop_df <- test_df |> dropp("s:")

    expect_true(!all(c("state", "sex") %in% names(drop_df)))
})


test_that("Drop variables ending with letter", {
    drop_df <- test_df |> dropp(":id")

    expect_true(!all(c("household_id", "person_id") %in% names(drop_df)))
})


test_that("Drop variables containing letter", {
    drop_df <- test_df |> dropp(":on:")

    expect_true(!all(c("person_id", "first_person", "education") %in% names(drop_df)))
})


test_that("Variables to drop contain a variable name that isn't part of the data frame", {
    expect_message(drop_df1 <- test_df |> dropp(year, age, sex, cats, dogs), " ! WARNING: The provided variable to drop")
    drop_df2 <- test_df |> dropp(year, age, sex)

    expect_equal(ncol(drop_df1), ncol(test_df) - 3)
    expect_equal(ncol(drop_df2), ncol(test_df) - 3)

    expect_identical(drop_df1, drop_df2)

    expect_true(!all(c("year", "sex", "age") %in% names(drop_df1)))
    expect_true(!all(c("cats", "dogs") %in% names(test_df)))
})


test_that("Drop only variables that are not part of the data frame", {
    expect_message(drop_df <- test_df |> dropp(cats, dogs), " ! WARNING: The provided variable to drop")

    expect_identical(drop_df, test_df)
})


test_that("Drop without any variables provided", {
    drop_df <- test_df |> dropp()

    expect_identical(drop_df, test_df)
})