tests/testthat/test-getUngaugedCluster.R

In inrae/airGRiwrm: 'airGR' Integrated Water Resource Management

test_that("Reservoir between ungauged and gauged node should be in ungauged cluster", {
  # Reservoir between Ungauged and gauged nodes
  n_rsrvr$model[n_rsrvr$id == "54095"] <- "Ungauged"
  g <- CreateGRiwrm(n_rsrvr) # Network provided by helper_RunModel_Reservoir.R
  expect_equal(unique(g$donor), "54001")
})

test_that("Reservoir supplied by derivated ungauged node should be included in the ungauged cluster", {
  # Reservoir between Ungauged and gauged nodes
  g <- CreateGRiwrm(n_derived_rsrvr) # Network provided by helper_RunModel_Reservoir.R
  expect_equal(g$donor[g$id == "Dam"], "54001")
  g2 <- getUngaugedCluster(g, "54001")
  expect_equal(g2$id, c("54095", "54095", "Dam", "54001"))
  expect_equal(g2$model[g2$id == "Dam"], "RunModel_Reservoir")
})


test_that("Reservoir and Diversion on reservoir should be in ungauged cluster", {
  nodes <- n_rsrvr
  nodes[nodes$id == "Dam", c("down", "length")] <- NA
  nodes$model[nodes$id == "54095"] <- "Ungauged"
  nodes <- rbind(nodes,
                 data.frame(id = "Dam", down = "54001", length = 42, area = NA, model = "Diversion"))
  g <- CreateGRiwrm(nodes)
  g2 <- getUngaugedCluster(g, "54001")
  expect_equal(g2$id,
               c("54095", "Dam", "Dam", "54001"))
})