tests/testthat/test-wholefunction_multiquarter.R

In dkgaraujo/brazilianbanks: Data on Brazilian banks and other regulated financial institutions

test_that("the main function works as a whole for a multi-quarter period without any adjustments", {

  qtrs <- brazilianbanks::all_available_quarters()
  num_qtrs <- length(qtrs)
  halfpoint <- num_qtrs / 2
  yyyymm_start <- sample(qtrs[1:halfpoint], 1)
  yyyymm_end <- sample(qtrs[halfpoint:num_qtrs], 1)

  dataset <- get_bank_stats(
    yyyymm_start = yyyymm_start,
    yyyymm_end = yyyymm_end,
    banks_only = FALSE,
    adjust_income_data = FALSE,
    include_growthrate = FALSE,
    cache_json = FALSE,
    verbose = FALSE)

  expect_gte(nrow(dataset), 50) # there are definitely more than 50 banks
})

test_that("the main function works as a whole for a multi-quarter period with the income data adjustment turned on", {
  # the income data adjustment depends on data from other quarters,
  # so it should recognise it has only one quarter and skip the adjustment

  qtrs <- brazilianbanks::all_available_quarters()
  num_qtrs <- length(qtrs)
  halfpoint <- num_qtrs / 2
  yyyymm_start <- sample(qtrs[1:halfpoint], 1)
  yyyymm_end <- sample(qtrs[halfpoint:num_qtrs], 1)

  dataset_with_adj <- get_bank_stats(
    yyyymm_start = yyyymm_start,
    yyyymm_end = yyyymm_end,
    banks_only = FALSE,
    adjust_income_data = TRUE,
    include_growthrate = FALSE,
    cache_json = FALSE,
    verbose = FALSE)

  dataset_without_adj <- get_bank_stats(
    yyyymm_start = yyyymm_start,
    yyyymm_end = yyyymm_end,
    banks_only = FALSE,
    adjust_income_data = FALSE,
    include_growthrate = FALSE,
    cache_json = FALSE,
    verbose = FALSE)

  expect_equal(dataset_with_adj, dataset_without_adj)
})

test_that("the main function works as a whole for a multi-quarter period with the growth rate adjustment", {
  # the growth rate adjustment depends on data from other quarters,
  # so it should recognise it has only one quarter and skip the adjustment

  qtrs <- brazilianbanks::all_available_quarters()
  num_qtrs <- length(qtrs)
  halfpoint <- num_qtrs / 2
  yyyymm_start <- sample(qtrs[1:halfpoint], 1)
  yyyymm_end <- sample(qtrs[halfpoint:num_qtrs], 1)

  dataset_with_adj <- get_bank_stats(
    yyyymm_start = yyyymm_start,
    yyyymm_end = yyyymm_end,
    banks_only = FALSE,
    adjust_income_data = FALSE,
    include_growthrate = TRUE,
    cache_json = FALSE,
    verbose = FALSE)

  dataset_without_adj <- get_bank_stats(
    yyyymm_start = yyyymm_start,
    yyyymm_end = yyyymm_end,
    banks_only = FALSE,
    adjust_income_data = FALSE,
    include_growthrate = FALSE,
    cache_json = FALSE,
    verbose = FALSE)

  expect_equal(dataset_with_adj, dataset_without_adj)
})


test_that("the main function works as a whole for a multi-quarter period with all adjustments", {

  qtrs <- brazilianbanks::all_available_quarters()
  num_qtrs <- length(qtrs)
  halfpoint <- num_qtrs / 2
  yyyymm_start <- sample(qtrs[1:halfpoint], 1)
  yyyymm_end <- sample(qtrs[halfpoint:num_qtrs], 1)

  dataset <- get_bank_stats(
    yyyymm_start = yyyymm_start,
    yyyymm_end = yyyymm_end,
    banks_only = FALSE,
    adjust_income_data = TRUE,
    include_growthrate = TRUE,
    cache_json = FALSE,
    verbose = FALSE)

  expect_gte(nrow(dataset), 50) # there are definitely more than 50 banks
})