test fixes

Author: dblodgett-usgs

R/collapse.R

@@ -26,6 +26,8 @@ collapse_flowlines <- function(flines, thresh, add_category = FALSE,
 
   check_names(flines, "collapse_flowlines")
 
+  flines[["toCOMID"]][flines[["toCOMID"]] == 0] <- NA
+  
   # very large thresh
   if (is.null(mainstem_thresh)) {
     mainstem_thresh_use <- max(flines$LENGTHKM)

tests/testthat/test_collapse.R

@@ -1,7 +1,8 @@
 context("collapse_flowlines")
 
 test_that("collapse flowlines works as expected", {
-  flines <- readRDS("data/petapsco_network.rds")
+  flines <- readRDS(list.files(pattern = "petapsco_network.rds", 
+                               full.names = TRUE, recursive = TRUE))
   flines <- sf::st_set_geometry(flines, NULL)
   flines <- suppressWarnings(nhdplusTools::prepare_nhdplus(flines, 20, 1))
   flines_out <- collapse_flowlines(flines, 1)