tests/testthat/test_aggregation.R

In ilangurudev/approxmapR: ApproxMAP (APPROXimate Multiple Alignment Pattern)

library(approxmapR)
library(purrr)
library(dplyr)
library(tidyr)
library(readr)

context("Aggregate Data")

inp <- read_csv("../../data/inp.csv")

# test_that("Units of time are converted to days properly",{
#   expect_equal(get_n_days("day", 5), 5)
#   expect_equal(get_n_days("week", 5), 35)
#   expect_equal(get_n_days("month", 5), 150)
#   expect_equal(get_n_days("6 months", 5), 900)
#   expect_equal(get_n_days("year", 5), 5*365)
# })


test_that("Convert to sequence returns a list", {
  expect_equal(typeof(inp %>% aggregate_sequences(format = "%Y-%m-%d") %>%
                      convert_to_sequence() %>%
                      pull(sequence)
                      ), "list")
})


inp2 <- inp %>%
  aggregate_sequences(format = "%Y-%m-%d") %>%
  convert_to_sequence() %>%
  pull(sequence)

inp3 <- inp %>%
  aggregate_sequences(format = "%Y-%m-%d", multiset = T) %>%
  convert_to_sequence() %>%
  pull(sequence)

test_that("Mutliset works", {
  expect_equal(inp2[[1]][[2]] %>% length(), 2)
  expect_equal(inp3[[1]][[2]] %>% length(), 3)
})



test_that("Preaggregation works", {
  t10 <- read_csv("../../data/t10.csv",
                  col_names = c("id", "period", "event"),
                  skip = 1, n_max = 150)

  expect_equal(suppressWarnings(t10 %>%
                                  get_weighted_sequence() %>%
                                  class()),
               "W_Sequence",
               fixed = TRUE)

  expect_warning(get_weighted_sequence(t10),
                 "Assuming that the dataframe you are using is pre-aggregated. If not, please use use the aggregate_sequences function.\n",
                 fixed = TRUE)

  names(t10) <- c("ID", "period", "event")
  expect_error(suppressWarnings(read_csv("../../data/t10.csv",
                          col_names = c("ID", "period", "event"),
                          skip = 1,  n_max = 150) %>%
               get_weighted_sequence()),
               "There should be 3 columns named id, period and event (all lower case).",
               fixed = TRUE)
})