tests/testthat/test-pivot_prep.R

In edgararuiz/pivotable: Create pivot tables

context("pivot_prep")

test_that("Pivot preparation works", {
  sales_pivot <- retail_orders %>%
    prep_dimensions(order_date = dim_hierarchy(
      year = as.integer(format(orderdate, "%Y")),
      month = as.integer(format(orderdate, "%m"))
    )) %>%
    prep_measures(no_orders = dplyr::n(), total_orders = sum(sales))
  expect_equal(class(sales_pivot), "pivot_prep")
  expect_output(print(sales_pivot))
  expect_output(print(pivot_values(sales_pivot, no_orders)))

  sales_pivot_2 <- retail_orders %>%
    prep_measures(no_orders = n(), total_orders = sum(sales))
  expect_equal(class(sales_pivot_2), "pivot_prep")

  sales_pivot_3 <- retail_orders %>%
    prep_dimensions(order_date = dim_hierarchy(
      year = as.integer(format(orderdate, "%Y")),
      month = as.integer(format(orderdate, "%m"))
    ))
  expect_equal(class(sales_pivot_3), "pivot_prep")
})

test_that("Pivot preparation works", {
  sales_pivot <- retail_orders %>%
    prep_dimensions(
      order_date = dim_hierarchy(
        year = as.integer(format(orderdate, "%Y")),
        month = as.integer(format(orderdate, "%m"))
      ),
      status
    ) %>%
    prep_measures(no_orders = dplyr::n(), total_orders = sum(sales)) %>%
    pivot_rows(order_date) %>%
    pivot_columns(status) %>%
    pivot_values(no_orders)

  expect_output(print(pivot_drill(sales_pivot, order_date)))
  expect_output(print(pivot_flip(sales_pivot)))
  expect_output(print(t(sales_pivot)))
  expect_output(print(
    pivot_totals(sales_pivot, include_column_totals = FALSE, include_row_totals = FALSE)
  ))
})