Add tests static coords

r-a-dobson · Jun 11, 2023 · e6cddf3 · e6cddf3
1 parent 8d8fbdc
commit e6cddf3
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 2 deletions.
diff --git a/R/extract_static_coords.R b/R/extract_static_coords.R
@@ -173,7 +173,7 @@ extract_static_coords <- function(occ.data,
       extracted_data <- cbind(extracted_data,
                               terra::extract(focal_raster,
                               y = as.matrix(occ.data[, c("x", "y")]),
-                              method = extraction.method )[,1])
+                              method = extraction.method[x] )[,1])
 
     }
 
@@ -190,7 +190,7 @@ extract_static_coords <- function(occ.data,
       extracted_data <- cbind(extracted_data,
                               terra::extract(static.rasters[[x]],
                                              y = as.matrix(occ.data[, c("x", "y")]),
-                                             method = extraction.method)[,1])
+                                             method = extraction.method[x])[,1])
     }
 
   }

diff --git a/tests/testthat/test-extract_static_coords.R b/tests/testthat/test-extract_static_coords.R
@@ -0,0 +1,32 @@
+data("sample_explan_data")
+random_cat_layer <- terra::rast(sample_extent_data)
+random_cat_layer <- terra::setValues(random_cat_layer,
+                                     sample(0:10, terra::ncell(random_cat_layer),
+                                          replace = TRUE))
+
+test_that("Works if static rasters and buffering", {
+results<-extract_static_coords(occ.data = sample_explan_data,
+                      varnames = "random_cat_layer",
+                      static.rasters = random_cat_layer)
+  expect_equal(ncol(results), ncol(sample_explan_data)+1)
+})
+
+
+test_that("Works if static rastersx2 and buffering", {
+  results<-extract_static_coords(occ.data = sample_explan_data,
+                                 varnames = c("random_cat_layer","random_cat_layer2"),
+                                 static.rasters = c(random_cat_layer,random_cat_layer))
+  expect_equal(ncol(results), ncol(sample_explan_data)+2)
+})
+
+
+
+test_that("Works if static rastersx2 and buffering", {
+  results<-extract_static_coords(occ.data = sample_explan_data,
+                                 varnames = c("random_cat_layer","random_cat_layer2"),
+                                 static.rasters = c(random_cat_layer,random_cat_layer),
+                                 moving.window.matrix = matrix(1,3,3),
+                                 GEE.math.fun = "sum")
+  expect_equal(ncol(results), ncol(sample_explan_data)+2)
+})
+