129 lines
2.7 KiB
C++
129 lines
2.7 KiB
C++
//
|
|
// test_epanet_toolkit.cpp
|
|
//
|
|
// Date Created: January 24, 2018
|
|
//
|
|
// Author: Michael E. Tryby
|
|
// US EPA - ORD/NRMRL
|
|
//
|
|
|
|
//#define BOOST_TEST_DYN_LINK
|
|
|
|
#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_alloc_free)
|
|
{
|
|
int error = 0;
|
|
EN_ProjectHandle ph = NULL;
|
|
|
|
error = EN_alloc(&ph);
|
|
|
|
BOOST_REQUIRE(error == 0);
|
|
BOOST_CHECK(ph != NULL);
|
|
|
|
error = EN_free(&ph);
|
|
|
|
BOOST_REQUIRE(error == 0);
|
|
BOOST_CHECK(ph == NULL);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE (test_open_close)
|
|
{
|
|
EN_ProjectHandle ph = NULL;
|
|
EN_alloc(&ph);
|
|
|
|
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);
|
|
|
|
int error = EN_open(ph, path_inp.c_str(), path_rpt.c_str(), path_out.c_str());
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
error = EN_close(ph);
|
|
BOOST_REQUIRE(error == 0);
|
|
|
|
EN_free(&ph);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_epanet)
|
|
{
|
|
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);
|
|
|
|
int error = ENepanet(path_inp.c_str(), path_rpt.c_str(), path_out.c_str(), NULL);
|
|
BOOST_REQUIRE(error == 0);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|
|
|
|
|
|
struct Fixture{
|
|
Fixture() {
|
|
|
|
path_inp = std::string(DATA_PATH_INP);
|
|
path_rpt = std::string(DATA_PATH_RPT);
|
|
path_out = std::string(DATA_PATH_OUT);
|
|
|
|
EN_alloc(&ph);
|
|
error = EN_open(ph, path_inp.c_str(), path_rpt.c_str(), path_out.c_str());
|
|
|
|
}
|
|
|
|
~Fixture() {
|
|
error = EN_close(&ph);
|
|
EN_free(&ph);
|
|
}
|
|
|
|
std::string path_inp;
|
|
std::string path_rpt;
|
|
std::string path_out;
|
|
|
|
int error;
|
|
EN_ProjectHandle ph;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// int main(int argc, char *argv[])
|
|
// {
|
|
// int error = 0;
|
|
// EN_ProjectHandle project = NULL;
|
|
|
|
// std::string data_path = std::string(PROJECT_HOME) + std::string(DATA_PATH);
|
|
// std::string inputFile(data_path);
|
|
// inputFile.append(std::string("/net1.inp"));
|
|
// std::string reportFile(data_path);
|
|
// reportFile.append(std::string("/net1.rpt"));
|
|
// std::string outputFile(data_path);
|
|
// outputFile.append(std::string("/net1.out"));
|
|
|
|
// error = EN_alloc(&project);
|
|
// error = EN_open(project, inputFile.c_str(), reportFile.c_str(), outputFile.c_str());
|
|
|
|
// error = EN_solveH(project);
|
|
// error = EN_solveQ(project);
|
|
// error = EN_report(project);
|
|
|
|
// error = EN_close(project);
|
|
// error = EN_free(&project);
|
|
|
|
// return error;
|
|
// }
|