72 lines
2.2 KiB
C++
72 lines
2.2 KiB
C++
//
|
|
// test_demand_categories.cpp
|
|
//
|
|
|
|
/*
|
|
This is a test for the demand categories names get\set APIs
|
|
A demand category name is set, the network is saved, reopened and the new demand category name is checked.
|
|
*/
|
|
|
|
#define BOOST_TEST_MODULE "toolkit"
|
|
#include <boost/test/included/unit_test.hpp>
|
|
|
|
#include <string>
|
|
#include "epanet2.h"
|
|
|
|
// NOTE: Project Home needs to be updated to run unit test
|
|
#define DATA_PATH_INP "./net1.inp"
|
|
#define DATA_PATH_RPT "./test.rpt"
|
|
#define DATA_PATH_OUT "./test.out"
|
|
|
|
using namespace std;
|
|
|
|
BOOST_AUTO_TEST_SUITE (test_toolkit)
|
|
|
|
BOOST_AUTO_TEST_CASE(test_demand_categories)
|
|
{
|
|
int error = 0;
|
|
int Nindex, ndem;
|
|
char demname[80];
|
|
EN_ProjectHandle ph = NULL;
|
|
|
|
std::string path_inp = std::string(DATA_PATH_INP);
|
|
std::string path_rpt = std::string(DATA_PATH_RPT);
|
|
std::string path_out = std::string(DATA_PATH_OUT);
|
|
|
|
error = EN_createproject(&ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_open(ph, path_inp.c_str(), path_rpt.c_str(), path_out.c_str());
|
|
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_REQUIRE(ndem == 1);
|
|
error = EN_setdemandname(ph, Nindex, ndem, (char *)"Demand category name");
|
|
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);
|
|
|
|
error = EN_createproject(&ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_open(ph, "net1_dem_cat.inp", path_rpt.c_str(), path_out.c_str());
|
|
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_REQUIRE(ndem == 1);
|
|
error = EN_getdemandname(ph, Nindex, ndem, demname);
|
|
BOOST_REQUIRE(error == 0);
|
|
BOOST_REQUIRE(strcmp(demname, "Demand category name")==0);
|
|
error = EN_close(ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
error = EN_deleteproject(&ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END() |