| 1 | context("Check that computeFilaments behaves as expected") |
| 2 | |
| 3 | #MOCK data; NOTE: could be in inst/testdata as well |
| 4 | getDataTest = function(n, shift) |
| 5 | { |
| 6 | x = seq(0,10,0.1) |
| 7 | L = length(x) |
| 8 | s1 = cos(x) |
| 9 | s2 = sin(x) |
| 10 | s3 = c( s1[1:(L%/%2)] , s2[(L%/%2+1):L] ) |
| 11 | #sum((s1-s2)^2) == 97.59381 |
| 12 | #sum((s1-s3)^2) == 57.03051 |
| 13 | #sum((s2-s3)^2) == 40.5633 |
| 14 | s = list( s1, s2, s3 ) |
| 15 | series = list() |
| 16 | for (i in seq_len(n)) |
| 17 | { |
| 18 | index = (i-1) %% 3 + 1 #map 1->1, 2->2, 3->3, 4->1, ..., 149->2, 150->3 |
| 19 | level = mean(s[[index]]) |
| 20 | serie = s[[index]] - level + rnorm(L,sd=0.01) |
| 21 | # 10 series with NAs for index 2 |
| 22 | if (index == 2 && i >= 60 && i<= 90) |
| 23 | serie[sample(seq_len(L),1)] = NA |
| 24 | series[[i]] = list("level"=level,"serie"=serie) #no need for more |
| 25 | } |
| 26 | if (shift) |
| 27 | { |
| 28 | # Simulate shift at origin when predict_at > 0 |
| 29 | series[2:(n+1)] = series[1:n] |
| 30 | series[[1]] = list("level"=0, "serie"=s[[1]][1:(L%/%2)]) |
| 31 | } |
| 32 | new("Data", data=series) |
| 33 | } |
| 34 | |
| 35 | test_that("output is as expected on simulated series, predict_at == 0", |
| 36 | { |
| 37 | data = getDataTest(150, shift=FALSE) |
| 38 | |
| 39 | # index 143 : serie type 2 |
| 40 | f = computeFilaments(data, 143, limit=60, plot=FALSE) |
| 41 | # Expected output: 23 series of type 3 (closer), then 50-3-10 series of type 2, then index 143 |
| 42 | expect_identical(length(f$indices), as.integer(61)) #61 because result also contain "today" |
| 43 | expect_identical(length(f$colors), as.integer(61)) |
| 44 | for (i in 1:23) |
| 45 | { |
| 46 | expect_equal((f$indices[i]-1) %% 3 + 1, 3) |
| 47 | expect_match(f$colors[i], f$colors[1]) |
| 48 | } |
| 49 | for (i in 24:60) |
| 50 | { |
| 51 | expect_equal((f$indices[i]-1) %% 3 + 1, 2) |
| 52 | expect_match(f$colors[i], f$colors[24]) |
| 53 | } |
| 54 | expect_equal(f$indices[61], 143) |
| 55 | expect_match(f$colors[61], "#FF0000") #special color: current day in red |
| 56 | expect_match(f$colors[1], "#E*") |
| 57 | expect_match(f$colors[24], "#1*") |
| 58 | }) |
| 59 | |
| 60 | test_that("output is as expected on simulated series, predict_at > 0", |
| 61 | { |
| 62 | data = getDataTest(150, shift=TRUE) |
| 63 | |
| 64 | # index 142 : serie type 3 |
| 65 | f = computeFilaments(data, 142, limit=70, plot=FALSE) |
| 66 | # Expected output: 24 series of type 2 (closer) then 50-4 series of type 3, then 142 |
| 67 | expect_identical(length(f$indices), as.integer(71)) |
| 68 | expect_identical(length(f$colors), as.integer(71)) |
| 69 | expect_true(all(f$indices >= 2)) #no type-1 neighbors |
| 70 | for (i in 1:24) |
| 71 | { |
| 72 | # -1 + -1 because of the initial shift |
| 73 | expect_equal((f$indices[i]-2) %% 3 + 1, 2) |
| 74 | expect_match(f$colors[i], f$colors[1]) |
| 75 | } |
| 76 | for (i in 25:70) |
| 77 | { |
| 78 | expect_equal((f$indices[i]-2) %% 3 + 1, 3) |
| 79 | expect_match(f$colors[i], f$colors[25]) |
| 80 | } |
| 81 | expect_equal(f$indices[71], 142) |
| 82 | expect_match(f$colors[71], "#FF0000") |
| 83 | expect_match(f$colors[1], "#E*") |
| 84 | expect_match(f$colors[25], "#1*") |
| 85 | }) |
| 86 | |
| 87 | test_that("output is as expected on simulated series", |
| 88 | { |
| 89 | data = getDataTest(150, shift=TRUE) |
| 90 | |
| 91 | # index 143 : serie type 1 |
| 92 | f = computeFilaments(data, 143, limit=60, plot=FALSE) |
| 93 | # Expected output: 23 series of type 3 (closer), then 50-10-3 series of type 1, then 143 |
| 94 | # NOTE: -10 because only past days with no-NAs tomorrow => exclude type 1 in [60,90[ |
| 95 | expect_identical(length(f$indices), as.integer(61)) |
| 96 | expect_identical(length(f$colors), as.integer(61)) |
| 97 | expect_true(all(f$indices >= 2)) #first_day should be 2 |
| 98 | for (i in 1:23) |
| 99 | { |
| 100 | expect_equal(( (f$indices[i]-2) %% 3 ) + 1, 3) |
| 101 | expect_match(f$colors[i], f$colors[1]) |
| 102 | } |
| 103 | for (i in 24:60) |
| 104 | { |
| 105 | expect_equal(( (f$indices[i]-2) %% 3 ) + 1, 1) |
| 106 | expect_match(f$colors[i], f$colors[24]) |
| 107 | } |
| 108 | expect_equal(f$indices[61], 143) |
| 109 | expect_match(f$colors[61], "#FF0000") |
| 110 | expect_match(f$colors[1], "#E*") |
| 111 | expect_match(f$colors[24], "#1*") |
| 112 | }) |