tests/testthat/test-load_data.R

In kristian-bak/kb.yahoo: Simple data loading from yahoo

test_that("Load yahoo data works", {

  data <- load_data(ticker = "NOVO-B.CO", from = "2021-01-01")

  cols <- c("Ticker", "Date", "Open", "High", "Low", "Close", "Adjusted", "Change", "ChangeSinceStart", "Volume")

  ## Expecting identical column names
  expect_identical(names(data), cols)

  price <- data %>%
    dplyr::filter(Date == "2021-01-05") %>%
    dplyr::pull(Close) %>%
    round(2)

  ## Expecting a certain closing value in the first row
  expect_equal(price, 423.95)

  ## Expecting failure for unknown ticker
  expect_error(
    load_data(ticker = "Unknown", from = "2021-01-01"),
    "Load data failed for Unknown"
  )

})

test_that("Load yahoo data with modified dates works", {

  data <- load_data(ticker = "TSLA", from = Sys.Date() - 7)

  expect_equal(class(data)[1], "tbl_df")

  expect_equal(names(data), c("Ticker", "Date", "Open", "High", "Low", "Close", "Adjusted", "Change",
                              "ChangeSinceStart", "Volume"))

})

test_that("Load FRED data works", {

  data <- load_data(ticker = "MORTGAGE30US", src = "FRED")

  cols <- c("Ticker", "Date", "MORTGAGE30US", "MORTGAGE30US_Change", "MORTGAGE30US_ChangeSinceStart")

  ## Expecting identical column names
  expect_identical(names(data), cols)

  price <- data %>%
    dplyr::filter(Date == "2022-01-06") %>%
    dplyr::pull(MORTGAGE30US) %>%
    round(2)

  ## Expecting a certain closing value in the first row
  expect_equal(price, 3.22)

  data2 <- load_data(ticker = "MORTGAGE30US", src = "FRED", prefix = FALSE)

  cols <- c("Ticker", "Date", "Close", "Change", "ChangeSinceStart")

  ## Expecting identical column names
  expect_identical(names(data2), cols)

})

test_that("Load multiple stocks work", {

  data <- load_data(ticker = c("NOVO-B.CO", "TSLA"), from = "2021-01-01")

  expect_equal(data %>%
    dplyr::pull(Ticker) %>%
    unique(),
    c("NOVO-B.CO", "TSLA")
  )

  expect_true(
    data %>%
      dplyr::slice(1) %>%
      dplyr::pull(Close) %>%
      is.numeric()
  )

})