New wrangling tests

Author: capnrefsmmat

R-packages/covidcast/tests/testthat/test-wrangle.R

@@ -0,0 +1,82 @@
+library(covidcast)
+library(dplyr)
+
+test_that("aggregate and shift", {
+  foo <- structure(data.frame(
+    data_source = "foo",
+    signal = "foo",
+    geo_value = "a",
+    value = 1:5,
+    time_value = seq.Date(as.Date("2020-01-01"), as.Date("2020-01-05"), "day"),
+    issue = as.Date("2020-01-06"),
+    stderr = 0.1,
+    sample_size = 10,
+    lag = 1),
+    class = c("covidcast_signal", "data.frame"))
+
+  agg <- aggregate_signals(foo, dt = c(-1, 1, 2))
+
+  expect_equal(arrange(agg, time_value),
+               structure(data.frame(
+                 geo_value = "a",
+                 time_value = foo$time_value,
+                 "value-1:foo_foo" = c(NA, 1:4),
+                 "value+1:foo_foo" = c(2:5, NA),
+                 "value+2:foo_foo" = c(3:5, NA, NA),
+                 check.names = FALSE),
+                 class = c("covidcast_signal_wide", "data.frame")))
+})
+
+test_that("widen and lengthen", {
+  foo <- structure(data.frame(
+    data_source = "foo",
+    signal = "foo",
+    geo_value = c("pa", "tx", "ri"),
+    value = 1:3,
+    time_value = as.Date("2020-01-01"),
+    issue = as.Date("2020-01-02"),
+    stderr = 0.5,
+    sample_size = 10,
+    lag = 1),
+    class = c("covidcast_signal", "data.frame"))
+
+  bar <- structure(data.frame(
+    data_source = "bar",
+    signal = "bar",
+    geo_value = c("pa", "tx", "ri"),
+    value = 4:6,
+    time_value = as.Date("2020-01-01"),
+    issue = as.Date("2020-01-02"),
+    stderr = 0.5,
+    sample_size = 10,
+    lag = 1),
+    class = c("covidcast_signal", "data.frame"))
+
+  agg_wide <- aggregate_signals(list(foo, bar), format = "wide")
+
+  expect_equal(arrange(agg_wide, geo_value),
+               structure(data.frame(
+                 geo_value = c("pa", "ri", "tx"),
+                 time_value = as.Date("2020-01-01"),
+                 "value+0:foo_foo" = c(1, 3, 2),
+                 "value+0:bar_bar" = c(4, 6, 5),
+                 check.names = FALSE),
+                 class = c("covidcast_signal_wide", "data.frame")))
+
+  long <- covidcast_longer(agg_wide)
+
+  expect_equal(arrange(long, data_source, geo_value),
+               structure(data.frame(
+                 data_source = c(rep("bar", 3), rep("foo", 3)),
+                 signal = c(rep("bar", 3), rep("foo", 3)),
+                 geo_value = rep(c("pa", "ri", "tx"), 2),
+                 time_value = as.Date("2020-01-01"),
+                 dt = 0,
+                 value = c(4, 6, 5, 1, 3, 2)),
+                 class = c("covidcast_signal_long", "data.frame")))
+
+  # Now try it long in the first place. Currently fails because wide format does
+  # not preserve issue, stderr, sample_size, or lag.
+  ## agg_long <- aggregate_signals(list(foo, bar), format = "long")
+  ## expect_equal(long, agg_long)
+})