inst/tinytest/test_wbstats.R

In wbstats: Programmatic Access to Data and Statistics from the World Bank API

library(wbstats)

if (requireNamespace("vcr", quietly = TRUE)) {

  vcr::vcr_configure(dir = "_vcr")

  # Focused test using the recorded cassette to validate basic contract of wb_data
  vcr::use_cassette(name = "wb_gdp_bra", {
    df <- wb_data("NY.GDP.MKTP.CD", country = "BRA", start_date = 2006, end_date = 2010)

    expect_inherits(df, "data.frame")
    expect_true(nrow(df) > 0)
    expect_true(is.numeric(df$date) || inherits(df$date, "Date"))

    expect_true(all(c("iso3c", "country", "date") %in% names(df)))
    expect_true("NY.GDP.MKTP.CD" %in% names(df))

    if (is.numeric(df$date)) {
      expect_equal(min(df$date, na.rm = TRUE), 2006)
      expect_equal(max(df$date, na.rm = TRUE), 2010)
    } else {
      yrs <- as.integer(format(df$date, "%Y"))
      expect_equal(min(yrs, na.rm = TRUE), 2006)
      expect_equal(max(yrs, na.rm = TRUE), 2010)
    }

    expect_true(all(unique(na.omit(df$iso3c)) == "BRA"))
  })

  # gdp for multiple countries - example
  vcr::use_cassette(name = "wb_gdp_multi", {
    df <- wb_data("NY.GDP.MKTP.CD", country = c("brazil", "mexico", "chile"), start_date = 2006)
    expect_inherits(df, "data.frame")
    expect_true(nrow(df) > 0)
    expect_true("NY.GDP.MKTP.CD" %in% names(df) || "value" %in% names(df))
  })

  # Brazil GDP all dates - example
  vcr::use_cassette(name = "wb_brazil_all", {
    df <- wb_data("NY.GDP.MKTP.CD", country = "br")
    expect_inherits(df, "data.frame")
    expect_true(all(unique(na.omit(df$iso3c)) == "BRA"))
  })

  # Brazil GDP 2006 - example
  vcr::use_cassette(name = "wb_brazil_2006", {
    df <- wb_data("NY.GDP.MKTP.CD", country = "brazil", start_date = 2006)
    expect_inherits(df, "data.frame")
    yrs <- if (is.numeric(df$date)) df$date else as.integer(format(df$date, "%Y"))
    expect_true(all(yrs >= 2006))
  })

  # multiple indicators (wide) - example
  vcr::use_cassette(name = "wb_multi_indicators", {
    # my_indicators <- c("SP.POP.TOTL", "NY.GDP.MKTP.CD", "SL.UEM.TOTL.ZS", "SP.DYN.CBRT.IN")
    my_indicators <- c("SP.POP.TOTL", "NY.GDP.MKTP.CD")
    # df <- wb_data(my_indicators)
    df <- wb_data(my_indicators, country = c("brazil", "chile", "mexico"), start_date = 2006)
    expect_inherits(df, "data.frame")
    expect_true(any(c("SP.POP.TOTL", "NY.GDP.MKTP.CD") %in% names(df)))
  })

  # multiple country ids mixed types - example
  vcr::use_cassette(name = "wb_multi_countries", {
    my_indicators <- c("SP.POP.TOTL", "NY.GDP.MKTP.CD")
    my_countries <- c("AL", "Geo", "mongolia")
    df <- wb_data(my_indicators, country = my_countries, start_date = 2005, end_date = 2007)
    expect_inherits(df, "data.frame")
    expect_true(nrow(df) > 0)
  })

  # long format return_wide = FALSE - example
  vcr::use_cassette(name = "wb_long_format", {
    my_indicators <- c("SP.POP.TOTL", "NY.GDP.MKTP.CD")
    my_countries <- c("AL", "Geo", "mongolia")
    df <- wb_data(my_indicators, country = my_countries, start_date = 2005, end_date = 2007, return_wide = FALSE)
    expect_inherits(df, "data.frame")
    expect_true(all(c("indicator_id", "value", "date") %in% names(df)))
  })

  # regions only - example
  vcr::use_cassette(name = "wb_region_totals", {
    df <- wb_data("SP.POP.TOTL", country = "regions_only", start_date = 2010, end_date = 2014)
    expect_inherits(df, "data.frame")
    expect_true(nrow(df) > 0)
  })

  # world aggregate - example
  vcr::use_cassette(name = "wb_world", {
    df <- wb_data("SP.POP.TOTL", country = "world", start_date = 2010, end_date = 2014)
    expect_inherits(df, "data.frame")
    expect_true(nrow(df) > 0)
  })

  # named indicators become column names (wide) - example
  vcr::use_cassette(name = "wb_named_vector_wide", {
    my_indicators <- c("pop" = "SP.POP.TOTL", "gdp" = "NY.GDP.MKTP.CD",
                        "unemployment_rate" = "SL.UEM.TOTL.ZS", "birth_rate" = "SP.DYN.CBRT.IN")
    df <- wb_data(my_indicators, country = "world", start_date = 2010, end_date = 2014)
    expect_inherits(df, "data.frame")
    expect_true(all(c("pop", "gdp") %in% names(df)))
  })

  # named indicators ignored in long format - example
  vcr::use_cassette(name = "wb_named_vector_long", {
    my_indicators <- c("pop" = "SP.POP.TOTL", "gdp" = "NY.GDP.MKTP.CD")
    df <- wb_data(my_indicators, country = "world", start_date = 2010, end_date = 2014, return_wide = FALSE)
    expect_inherits(df, "data.frame")
    expect_true(all(c("indicator_id", "indicator") %in% names(df)))
  })

  # named indicators long format with language - example
  vcr::use_cassette(name = "wb_named_vector_long_bg", {
    my_indicators <- c("pop" = "SP.POP.TOTL", "gdp" = "NY.GDP.MKTP.CD")
    df <- wb_data(my_indicators, country = "world", start_date = 2010, end_date = 2014,
                  return_wide = FALSE, lang = "bg")
    expect_inherits(df, "data.frame")
    expect_true("indicator" %in% names(df))
  })
}