tests/testthat/test-parse_time_series.R

In SawickiBartosz/tidycharts: Generate Tidy Charts Inspired by 'IBCS'

context('Time series parsing')

test_that("dummy months example works", {
  df <- data.frame(x = as.Date(c(
    '2021-01-01', '2021-02-01', '2021-03-01'
  )),
  y = c(10, 12, 15))
  result <- parse_time_series_column(df = df, x = 'x', y = 'y')
  expected <- data.frame(
    x = c(3, 3, 3),
    y = c(10, 12, 15),
    cat = paste(month.abb[1:3])
  )
  expect_equal(result, expected, tolerance = 0.12)
})

test_that('parsing into quarters works', {
  df <- data.frame(x = as.Date(c(
    '2021-07-19', '2021-07-18', '2021-07-24'
  )),
  y = c(10, 21, 15))
  result <-
    parse_time_series_column(
      df = df,
      x = 'x',
      y = 'y',
      convert.to = 'quarters'
    )
  expected <- data.frame(
    x = c(19 / 92, 18 / 92, 24/ 92) * 100,
    y = c(10, 21, 15),
    cat = c('Q3', 'Q3', 'Q3')
  )
  expect_equal(result, expected)
})



test_that('parsing into quarters works', {
  df <- data.frame(x = as.Date(c(
    '2021-01-01', '2021-01-31', '2021-12-31'
  )),
  y = c(10, 21, 15))
  result <-
    parse_time_series_column(
      df = df,
      x = 'x',
      y = 'y',
      convert.to = 'years'
    )
  expected <- data.frame(
    x = c(1 / 365, 31 / 365, 365/ 365) * 100,
    y = c(10, 21, 15),
    cat = c(2021, 2021, 2021)
  )
  expect_equal(result, expected)
})


test_that('parsing into weeks works', {
  df <- data.frame(x = as.Date(c(
    '2021-07-19', '2021-07-18', '2021-07-24'
  )),
  y = c(10, 21, 15))
  result <-
    parse_time_series_column(
      df = df,
      x = 'x',
      y = 'y',
      convert.to = 'weeks'
    )
  expected <- data.frame(
    x = c(2 / 7, 1 / 7, 1) * 100,
    y = c(10, 21, 15),
    cat = c(29, 29, 29)
  )
  expect_equal(result, expected)
})

test_that("wrapping function works", {
  df <- data.frame(
    x = as.Date(c(
      '2021-01-01', '2021-02-01', '2021-03-01'
    )),
    y1 = c(10, 12, 15),
    y2 = c(11, 10, 14),
    y3 = c(12, 12, 12)
  )
  result <-
    parse_time_series(df = df,
                      dates = 'x',
                      series = c('y1', 'y2', 'y3'))

  expected <- list(
    data.frame(
      x = c(3, 3, 3),
      y = c(10, 12, 15),
      cat = paste(month.abb[1:3])
    ),
    data.frame(
      x = c(3, 3, 3),
      y = c(11, 10, 14),
      cat = paste(month.abb[1:3])
    ),
    data.frame(
      x = c(3, 3, 3),
      y = c(12, 12, 12),
      cat = paste(month.abb[1:3])
    )

  )
  expect_equal(result, expected, tolerance = 0.12)
})