153 lines
4.0 KiB
C++
153 lines
4.0 KiB
C++
/*
|
|
******************************************************************************
|
|
Project: OWA EPANET
|
|
Version: 2.2
|
|
Module: test_demand.cpp
|
|
Description: Tests EPANET toolkit api functions
|
|
Authors: see AUTHORS
|
|
Copyright: see AUTHORS
|
|
License: see LICENSE
|
|
Last Updated: 03/21/2019
|
|
******************************************************************************
|
|
*/
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
#include "test_toolkit.hpp"
|
|
|
|
|
|
BOOST_AUTO_TEST_SUITE (test_demand)
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(test_categories_save)
|
|
{
|
|
int error = 0;
|
|
int Nindex, ndem;
|
|
|
|
EN_Project ph = NULL;
|
|
|
|
error = EN_createproject(&ph);
|
|
error = EN_open(ph, DATA_PATH_NET1, DATA_PATH_RPT, DATA_PATH_OUT);
|
|
|
|
error = EN_getnodeindex(ph, (char *)"12", &Nindex);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_getnumdemands(ph, Nindex, &ndem);
|
|
BOOST_REQUIRE(error == 0);
|
|
BOOST_CHECK(ndem == 1);
|
|
|
|
char demname[31];
|
|
error = EN_getdemandname(ph, Nindex, ndem, demname);
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_setdemandname(ph, Nindex, ndem, (char *)"CUB_SCOUT_MOTOR_POOL");
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_saveinpfile(ph, "net1_dem_cat.inp");
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_close(ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_deleteproject(ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
}
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(test_categories_reopen, * boost::unit_test::depends_on("test_demand/test_categories_save"))
|
|
{
|
|
int error = 0;
|
|
int Nindex, ndem;
|
|
|
|
EN_Project ph = NULL;
|
|
|
|
BOOST_TEST_CHECKPOINT("Reopening saved input file");
|
|
error = EN_createproject(&ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_open(ph, "net1_dem_cat.inp", DATA_PATH_RPT, DATA_PATH_OUT);
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_getnodeindex(ph, (char *)"12", &Nindex);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_getnumdemands(ph, Nindex, &ndem);
|
|
BOOST_REQUIRE(error == 0);
|
|
BOOST_CHECK(ndem == 1);
|
|
|
|
char demname[31];
|
|
error = EN_getdemandname(ph, Nindex, ndem, demname);
|
|
BOOST_CHECK(error == 0);
|
|
|
|
BOOST_CHECK(check_string(demname, "CUB_SCOUT_MOTOR_POOL"));
|
|
|
|
error = EN_close(ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_deleteproject(ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
}
|
|
|
|
BOOST_FIXTURE_TEST_CASE(test_adddemand, FixtureSingleNode)
|
|
{
|
|
int Dindex, nD1, nD2;
|
|
|
|
error = EN_adddemand(ph, node_qhut, 100.0, "PrimaryPattern", "PrimaryDemand");
|
|
BOOST_CHECK(error != 0);
|
|
|
|
error = EN_addpattern(ph, (char *)"PrimaryPattern");
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_adddemand(ph, node_qhut, 100.0, "PrimaryPattern", "PrimaryDemand");
|
|
BOOST_CHECK(error == 0);
|
|
|
|
error = EN_addpattern(ph, (char *)"SecondaryPattern");
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_adddemand(ph, node_qhut, 10.0, "SecondaryPattern", "SecondaryDemand");
|
|
BOOST_CHECK(error == 0);
|
|
|
|
error = EN_addpattern(ph, (char *)"TertiaryPattern");
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_adddemand(ph, node_qhut, 1.0, "TertiaryPattern", "TertiaryDemand");
|
|
BOOST_CHECK(error == 0);
|
|
|
|
error = EN_getnumdemands(ph, node_qhut, &nD1);
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_getdemandindex(ph, node_qhut, "TertiaryDemand", &Dindex);
|
|
BOOST_CHECK(error == 0);
|
|
BOOST_CHECK(Dindex == nD1);
|
|
|
|
error = EN_deletedemand(ph, node_qhut, Dindex);
|
|
BOOST_CHECK(error == 0);
|
|
|
|
error = EN_getnumdemands(ph, node_qhut, &nD2);
|
|
BOOST_REQUIRE(error == 0);
|
|
BOOST_CHECK(nD1 - nD2 == 1);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_cms_unit)
|
|
{
|
|
int flowType;
|
|
|
|
EN_Project ph = NULL;
|
|
EN_createproject(&ph);
|
|
|
|
int error = EN_init(ph, DATA_PATH_RPT, DATA_PATH_OUT, EN_CMS, EN_HW);
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_setflowunits(ph, EN_CMS);
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_getflowunits(ph, &flowType );
|
|
BOOST_REQUIRE(error == 0);
|
|
BOOST_REQUIRE(flowType == EN_CMS);
|
|
|
|
error = EN_close(ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
int error = EN_init(ph, DATA_PATH_RPT, DATA_PATH_OUT, EN_CMS+1, EN_HW);
|
|
BOOST_REQUIRE(error == 251);
|
|
|
|
EN_deleteproject(ph);
|
|
}
|
|
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|