context("Generic tests") #source("R/zzz.R") #source("R/SelectPeriodOnDates.R") #source("R/SelectPeriodOnData.R") library(s2dv) test_that("Decadal", { # -------- DECADAL ----------# # decadal: 1 sdate several consequtive years: dates <- seq(as.Date("01-01-2000", "%d-%m-%Y", tz = 'UTC'), as.Date("31-12-2005","%d-%m-%Y", tz = 'UTC'), "day") # No dims -> test .position output <- c( seq(as.Date("2000-02-01", "%Y-%m-%d"), as.Date("2000-02-10", "%Y-%m-%d"), 'day'), seq(as.Date("2001-02-01", "%Y-%m-%d"), as.Date("2001-02-10", "%Y-%m-%d"), 'day'), seq(as.Date("2002-02-01", "%Y-%m-%d"), as.Date("2002-02-10", "%Y-%m-%d"), 'day'), seq(as.Date("2003-02-01", "%Y-%m-%d"), as.Date("2003-02-10", "%Y-%m-%d"), 'day'), seq(as.Date("2004-02-01", "%Y-%m-%d"), as.Date("2004-02-10", "%Y-%m-%d"), 'day'), seq(as.Date("2005-02-01", "%Y-%m-%d"), as.Date("2005-02-10", "%Y-%m-%d"), 'day')) dim(output) <- c(ftime = 60) expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 2)), output) data <- array(1:(length(dates)*3), c(memb = 1, ftime = length(dates), lon = 3)) expect_equal( SelectPeriodOnData(data, dates, start = list(1, 2), end = list(10, 2)), array(c(c(32:41, 398:407, 763:772, 1128:1137, 1493:1502, 1859:1868), c(32:41, 398:407, 763:772, 1128:1137, 1493:1502, 1859:1868) + 2192, c(32:41, 398:407, 763:772, 1128:1137, 1493:1502, 1859:1868) + 2 * 2192), c(ftime = 60, memb = 1, lon = 3))) output2 <- c( seq(as.Date("2000-02-01", "%Y-%m-%d"), as.Date("2000-04-10", "%Y-%m-%d"), 'day'), seq(as.Date("2001-02-01", "%Y-%m-%d"), as.Date("2001-04-10", "%Y-%m-%d"), 'day'), seq(as.Date("2002-02-01", "%Y-%m-%d"), as.Date("2002-04-10", "%Y-%m-%d"), 'day'), seq(as.Date("2003-02-01", "%Y-%m-%d"), as.Date("2003-04-10", "%Y-%m-%d"), 'day'), seq(as.Date("2004-02-01", "%Y-%m-%d"), as.Date("2004-04-10", "%Y-%m-%d"), 'day'), seq(as.Date("2005-02-01", "%Y-%m-%d"), as.Date("2005-04-10", "%Y-%m-%d"), 'day')) dim(output2) <- c(ftime = 416) expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 4)), output2) expect_equal( SelectPeriodOnData(data, dates, start = list(1, 2), end = list(10, 4)), array(c(c(32:101, 398:466, 763:831, 1128:1196, 1493:1562, 1859:1927), c(32:101, 398:466, 763:831, 1128:1196, 1493:1562, 1859:1927) + 2192, c(32:101, 398:466, 763:831, 1128:1196, 1493:1562, 1859:1927) + 2 * 2192), c(ftime = 416, memb = 1, lon = 3))) # 1 dim -> test Apply dim(dates) <- c(ftime = length(dates)) expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 2)), output) # no need to check on Data, repited expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 4)), output2) # no need to check on Data, repited # decadal: 5 sdates several consequtive years dates <- rep(seq(as.Date("01-01-2000", "%d-%m-%Y", tz = 'UTC'), as.Date("31-12-2005","%d-%m-%Y", tz = 'UTC'), "day"), 5) dim(dates) <- c(ftime = 2192, sdate = 5) output3 <- rep(output, 5) dim(output3) <- c(ftime = 60, sdate = 5) expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 2)), output3) data <- array(1:(length(dates)*3), c(memb = 1, sdate = 5, ftime = length(dates)/5, lon = 3)) expect_equal( #To be extended for all sdate dimensions: SelectPeriodOnData(data, dates, start = list(1, 2), end = list(10, 2))[,1,1,1], c(1:10 * 5 + 151, 1:10 * 5 + 1981, 1:10 * 5 + 3806, 1:10 * 5 + 5631, 1:10 * 5 + 7456, 1:10 * 5 + 9286)) output4 <- rep(output2, 5) dim(output4) <- c(ftime = 416, sdate = 5) expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 4)), output4) expect_equal( #To be extended for all ftime dimensions: SelectPeriodOnData(data, dates, start = list(1, 2), end = list(10, 4))[1,1,,1], 156:160) # Multiple dims: sdate, fyear, ftime library(CSTools) dates <- SplitDim(dates, indices = dates[,1], split_dim = 'ftime', freq = 'year') dates <- as.POSIXct(dates * 24 * 3600, origin = '1970-01-01', tz = 'UTC') output5 <- SplitDim(output3, indices = output3[,1], split_dim = 'ftime' , freq = 'year') output5 <- as.POSIXct(output5 * 24 * 3600, origin = '1970-01-01', tz = 'UTC') expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 2)), output5) data <- array(1:(366*6*5*3), c(memb = 1, sdate = 5, year = 6, ftime = 366, lon = 3)) expect_equal( SelectPeriodOnData(data, dates, start = list(1, 2), end = list(10, 2)), InsertDim(Reorder(data[,,,32:41,], c('ftime', 'sdate', 'year', 'lon')), len = 1, pos = 2, name = 'memb')) output6 <- SplitDim(output4, indices = output4[,1], split_dim = 'ftime' , freq = 'year') output6 <- as.POSIXct(output6 * 24 * 3600, origin = '1970-01-01', tz = 'UTC') expect_equal( SelectPeriodOnDates(dates, start = list(1, 2), end = list(10, 4)), output6) #expect_equal( # to be fixed: # SelectPeriodOnData(data, dates, start = list(1, 2), end = list(10, 4)), # (931:935), outer(seq(931, 3001, 30), 0:4, '+') # InsertDim(Reorder(data[,,,32:41,], c('ftime', 'sdate', 'year', 'lon')), # len = 1, pos = 2, name = 'memb')) }) test_that("Seasonal", { # 1 start month, select the required 'ftime' of each 'sdate' in-between the entire timeseries dates <- c(seq(as.Date("01-04-2000", format = "%d-%m-%Y"), as.Date("31-10-2000", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("01-04-2001", format = "%d-%m-%Y"), as.Date("31-10-2001", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("01-04-2002", format = "%d-%m-%Y"), as.Date("31-10-2002", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("01-04-2003", format = "%d-%m-%Y"), as.Date("31-10-2003", format = "%d-%m-%Y"), by = 'day')) output <- c(seq(as.Date("21-04-2000", format = "%d-%m-%Y"), as.Date("21-06-2000", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("21-04-2001", format = "%d-%m-%Y"), as.Date("21-06-2001", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("21-04-2002", format = "%d-%m-%Y"), as.Date("21-06-2002", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("21-04-2003", format = "%d-%m-%Y"), as.Date("21-06-2003", format = "%d-%m-%Y"), by = 'day')) dim(output) <- c(ftime = (30 - 20 + 31 + 21) * 4) expect_equal( SelectPeriodOnDates(dates, start = list(21, 4), end = list(21, 6)), output) # following the above case, and select the data data <- array(1:(5 * 4 * 214 * 2), c(memb = 5, sdate = 4, ftime = 214, lon = 2)) dim(dates) <- c(ftime = 214, sdate = 4) expect_equal( SelectPeriodOnData(data, dates, start = list(21, 4), end = list(21, 6))[,1,1,1], data[1,1,21:82,1]) # when selecting the days across two years dates <- seq(as.Date("2000-01-01", "%Y-%m-%d"), as.Date("2003-12-31", "%Y-%m-%d"), 'day') output1 <- c(seq(as.Date("01-01-2000", format = "%d-%m-%Y"), as.Date("31-01-2000", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("01-12-2000", format = "%d-%m-%Y"), as.Date("31-01-2001", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("01-12-2001", format = "%d-%m-%Y"), as.Date("31-01-2002", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("01-12-2002", format = "%d-%m-%Y"), as.Date("31-01-2003", format = "%d-%m-%Y"), by = 'day'), seq(as.Date("01-12-2003", format = "%d-%m-%Y"), as.Date("31-12-2003", format = "%d-%m-%Y"), by = 'day')) dim(output1) <- c(ftime = 31 * 8) expect_equal( SelectPeriodOnDates(dates, start = list(1, 12), end = list(31, 1)), output1) # following the above case, and select the data data1 <- array(1:(length(dates) * 2), c(memb = 1, ftime = length(dates), lon = 2)) expect_equal( SelectPeriodOnData(data1, dates, start = list(1, 12), end = list(31, 1)), array(c(c(1:31, 336:397, 701:762, 1066:1127, 1431:1461), c(1:31, 336:397, 701:762, 1066:1127, 1431:1461) + 1461), c(ftime = 31 * 8, memb = 1, lon = 2))) })