tests/testthat/test-demo.R

In duckplyr: A 'DuckDB'-Backed Version of 'dplyr'

test_that("demo queries work", {
  skip_if_not_installed("lubridate")

  withr::local_envvar(DUCKPLYR_FORCE = TRUE)

  wday <- lubridate::wday
  hour <- lubridate::hour

  expect_snapshot({
    tips_by_day_hour <- taxi_data_2019 %>%
      filter(total_amount > 0) %>%
      mutate(
        tip_pct = 100 * tip_amount / total_amount,
        dn = wday(pickup_datetime),
        hr = hour(pickup_datetime)
      ) %>%
      summarise(
        avg_tip_pct = mean(tip_pct),
        n = n(),
        .by = c(dn, hr)
      ) %>%
      arrange(desc(avg_tip_pct))

    tips_by_day_hour
  })

  expect_snapshot({
    tips_by_passenger <- taxi_data_2019 %>%
      filter(total_amount > 0) %>%
      mutate(tip_pct = 100 * tip_amount / total_amount) %>%
      summarise(
        avg_tip_pct = median(tip_pct),
        n = n(),
        .by = passenger_count
      ) %>%
      arrange(desc(passenger_count))

    tips_by_passenger
  })

  expect_snapshot({
    popular_manhattan_cab_rides <- taxi_data_2019 %>%
      filter(total_amount > 0) %>%
      inner_join(zone_map, by = join_by(pickup_location_id == LocationID)) %>%
      inner_join(zone_map, by = join_by(dropoff_location_id == LocationID)) %>%
      filter(Borough.x == "Manhattan", Borough.y == "Manhattan") %>%
      select(start_neighborhood = Zone.x, end_neighborhood = Zone.y) %>%
      summarise(
        num_trips = n(),
        .by = c(start_neighborhood, end_neighborhood),
      ) %>%
      arrange(desc(num_trips))

    popular_manhattan_cab_rides
  })

  expect_snapshot({
    num_trips_per_borough <- taxi_data_2019 %>%
      filter(total_amount > 0) %>%
      inner_join(zone_map, by = join_by(pickup_location_id == LocationID)) %>%
      inner_join(zone_map, by = join_by(dropoff_location_id == LocationID)) %>%
      mutate(pickup_borough = Borough.x, dropoff_borough = Borough.y) %>%
      select(pickup_borough, dropoff_borough, tip_amount) %>%
      summarise(
        num_trips = n(),
        .by = c(pickup_borough, dropoff_borough)
      )

    num_trips_per_borough_no_tip <- taxi_data_2019 %>%
      filter(total_amount > 0, tip_amount == 0) %>%
      inner_join(zone_map, by = join_by(pickup_location_id == LocationID)) %>%
      inner_join(zone_map, by = join_by(dropoff_location_id == LocationID)) %>%
      mutate(
        pickup_borough = Borough.x,
        dropoff_borough = Borough.y,
        tip_amount
      ) %>%
      summarise(
        num_zero_tip_trips = n(),
        .by = c(pickup_borough, dropoff_borough)
      )

    num_zero_percent_trips <- num_trips_per_borough %>%
      inner_join(
        num_trips_per_borough_no_tip,
        by = join_by(pickup_borough, dropoff_borough)
      ) %>%
      mutate(
        num_trips = num_trips,
        percent_zero_tips_trips = 100 * num_zero_tip_trips / num_trips
      ) %>%
      select(
        pickup_borough,
        dropoff_borough,
        num_trips,
        percent_zero_tips_trips
      ) %>%
      arrange(desc(percent_zero_tips_trips))

    num_zero_percent_trips
  })
})