library(medfateland)

data("example_ifn")
data("example_watershed")
ypts = example_ifn
example_watershed$channel <- FALSE
example_watershed$channel[c(6, 11, 12, 20)] <- TRUE
yws = example_watershed
r <-terra::rast(xmin = 401380, ymin = 4671820, xmax = 402880, ymax = 4672620, 
                nrow = 8, ncol = 15, crs = "epsg:32631")

data("SpParamsMED")
vars <- c("elevation", "slope", "aspect", "land_cover_type",
          "soil_vol_extract", "soil_vol_sat", "soil_vol_fc", "soil_vol_wp", "soil_vol_curr",
          "soil_theta_curr", "soil_psi_curr", "soil_rwc_curr", "soil_rew_curr",
          "basal_area", "leaf_area_index", "foliar_biomass", "fuel", "shrub_volume")
vars_ws <- c(vars,
             "depth_to_bedrock","bedrock_porosity", "bedrock_conductivity",
             "aquifer_elevation", "depth_to_aquifer","aquifer", "snowpack", "channel")
test_that("Can extract variables from 'sf' object",{
  expect_s3_class(extract_variables(ypts, vars = vars, SpParams = SpParamsMED), "sf")
})

test_that("Can plot variables from point 'sf' object",{
  for(v in vars) expect_s3_class(plot_variable(ypts, variable = v, SpParams = SpParamsMED), "ggplot")
})

test_that("Can extract variables from watershed 'sf' object",{
  expect_s3_class(extract_variables(yws, vars = vars_ws, SpParams = SpParamsMED), "sf")
})

test_that("Can plot variables from watershed 'sf' object",{
  for(v in vars_ws) {
    expect_s3_class(plot_variable(yws, variable=v, SpParams = SpParamsMED), "ggplot")
  }
})

test_that("Overland routing works for watershed",{
  expect_type(cell_neighbors(r = r, sf = yws), "list")
  expect_s3_class(overland_routing(r = r, sf = yws), "sf")
})