tests/testthat/test_controls.R

In epanet2toolkit: Call 'EPANET' Functions to Simulate Pipe Networks

#*****************************************
#
# (C) Copyright IBM Corp. 2017, 2020
# Author: Ernesto Arandia & Bradley J Eck
#
#*****************************************

context("get control")
test_that("no crash calling on closed toolkit",{
  expect_error( x <- ENgetcontrol(1) ) 
})
test_that("works for single input (1)",{
  ENopen("Net3.inp","Net3.rpt")
  lidx <- 	ENgetcontrol(1)
  ENclose()
  expect_equal(lidx, list(ctype=2, lindex=118, setting=1, nindex=0, level=3600))
})
test_that("works for single input (2)",{
  ENopen("Net3.inp","Net3.rpt")
  lidx <- 	ENgetcontrol(5)
  ENclose()
  expect_equal(lidx, list(ctype=0, lindex=116, setting=0, nindex=95, level=17.1),
               tolerance=1e-7)
})
test_that("works for multiple input",{
  ENopen("Net3.inp","Net3.rpt")
  expect_error(lid <- ENgetcontrol(c(1,2,3)))
  ENclose()
})
test_that("get error 241 on no ID",{
  ENopen("Net3.inp","Net3.rpt")
  expect_error(ENgetcontrol(7),"241")
  ENclose()
})

context("set control")
test_that("no crash calling on closed toolkit",{
  expect_error( ENsetcontrol(1, ctype=2, lindex=118, setting=1, nindex=0, level=5400) )
})
test_that("works on single input",{
  ENopen("Net3.inp","Net3.rpt")
  ENsetcontrol(1, ctype=2, lindex=118, setting=1, nindex=0, level=5400)		
  x <- ENgetcontrol(1)
  ENclose()
  expect_equal(x, list(ctype=2, lindex=118, setting=1, nindex=0, level=5400), tolerance = 1e-7)
})
test_that("error on multiple indexes",{
  ENopen("Net3.inp","Net3.rpt")
  expect_error(ENsetcontrol(c(1,2), ctype=2, lindex=118, setting=1, nindex=0, level=5400))
  ENclose()
})
test_that("get error on wrong control",{
  ENopen("Net3.inp","Net3.rpt")
  expect_error(ENsetcontrol(7,1,116,1,95,21.5))
  ENclose()
})

context("delete control")
test_that("can delete existing control",{

  ENopen("Net3.inp", "Net3.rpt")
  x <- ENdeletecontrol(1)
  ENclose()
  expect_null(x)

})

context("add control")
test_that("can add new control",{

  ENopen("Net3.inp", "Net3.rpt")
  # pipe from river operates only part of the day
  lidx <- ENgetlinkindex("50")
  cidx <- ENaddcontrol(type="EN_TIMEOFDAY", linkIndex=lidx,setting=0, nodeIndex = 0, level=20000)
  ctrl <- ENgetcontrol(cidx)
  expect_equal(ctrl$lindex,lidx)
  ENclose()
})