Reorganizing unit tests

Unit tests for the toolkit are consolidated into one test module with a separate test suite for related API functions.
2019-03-21 13:11:10 -04:00
parent 80ce382a38
commit 5687c63548
17 changed files with 511 additions and 363 deletions
--- a/tests/test_pattern.cpp
+++ b/tests/test_pattern.cpp
@@ -0,0 +1,135 @@
+/*
+ ******************************************************************************
+ Project:      OWA EPANET
+ Version:      2.2
+ Module:       test_pattern.cpp
+ Description:  Tests EPANET toolkit api functions
+ Authors:      see AUTHORS
+ Copyright:    see AUTHORS
+ License:      see LICENSE
+ Last Updated: 03/21/2019
+ ******************************************************************************
+*/
+
+#define BOOST_ALL_DYN_LINK
+#include <boost/test/unit_test.hpp>
+
+#include "test_toolkit.hpp"
+
+
+BOOST_AUTO_TEST_SUITE (pattern)
+
+BOOST_AUTO_TEST_CASE(add_set_pattern)
+{
+	std::string path_inp(DATA_PATH_NET1);
+	std::string path_rpt(DATA_PATH_RPT);
+	std::string path_out(DATA_PATH_OUT);
+	std::string inp_save("net1_setid.inp");
+
+    int error = 0;
+
+    EN_Project ph = NULL;
+    EN_createproject(&ph);
+
+    error = EN_open(ph, path_inp.c_str(), path_rpt.c_str(), "");
+    BOOST_REQUIRE(error == 0);
+
+    // Get the default pattern index
+    double dblPatIdx;
+    int defPatIdx;
+    int patIdx;
+    EN_getoption(ph, EN_DEFDEMANDPAT, &dblPatIdx);
+    defPatIdx = (int)dblPatIdx;
+
+    // Rename the default pattern
+    EN_setpatternid(ph, defPatIdx, (char *)"Pat1");
+    EN_getpatternindex(ph, (char *)"Pat1", &patIdx);
+    BOOST_REQUIRE(defPatIdx == patIdx);
+
+    // Add 2 new patterns
+    EN_addpattern(ph, (char *)"Pat2");
+    EN_addpattern(ph, (char *)"Pat3");
+    double f2[] = {2.1, 2.2};
+    double f3[] = {3.1, 3.2, 3.3, 3.4};
+    EN_setpattern(ph, 2, f2, 2);
+    EN_setpattern(ph, 3, f3, 4);
+
+    // Assign Pat3 to 3rd junction
+    EN_setdemandpattern(ph, 3, 1, 3);
+
+    // Delete Pat2
+    EN_deletepattern(ph, 2);
+
+    // Check that there are now 2 patterns
+    int n;
+    EN_getcount(ph, EN_PATCOUNT, &n);
+    BOOST_REQUIRE(n == 2);
+
+    // Check that Pat3 with 4 factors is still assigned to 3rd junction
+    EN_getdemandpattern(ph, 3, 1, &patIdx);
+    EN_getpatternlen(ph, patIdx, &n);
+    BOOST_REQUIRE(n == 4);
+
+    // Delete the default pattern
+    EN_deletepattern(ph, defPatIdx);
+
+    // Check that junction 4 has no pattern
+    EN_getdemandpattern(ph, 4, 1, &patIdx);
+    BOOST_REQUIRE(patIdx == 0);
+
+    // And that junction 3 still uses Pat3
+    EN_getdemandpattern(ph, 3, 1, &patIdx);
+    char patID[EN_MAXID+1];
+    EN_getpatternid(ph, patIdx, patID);
+    BOOST_REQUIRE(strcmp(patID, "Pat3") == 0);
+
+    // Re-name the pump's (Link 9) head curve
+    int pumpIdx;
+    char pumpID[] = "9";
+    int curveIdx;
+    char newCurveID[] = "PumpHeadCurve";
+    EN_getlinkindex(ph, pumpID, &pumpIdx);
+    EN_getheadcurveindex(ph, pumpIdx, &curveIdx);
+    EN_setcurveid(ph, curveIdx, newCurveID);
+
+    // Check that rename was successful
+    char curveID[EN_MAXID + 1];
+    EN_getcurveid(ph, curveIdx, curveID);
+    BOOST_REQUIRE(strcmp(curveID, newCurveID) == 0);
+
+    // Add two new curves
+    double x2[] = {0.0, 1.0, 2.0};
+    double y2[] = {400.0, 60.0, 30.0};
+    double x3[] = {2000.0};
+    double y3[] = {100.0};
+    char curve2[] = "Curve2";
+    char curve3[] = "Curve3";
+    EN_addcurve(ph, curve2);
+    EN_setcurve(ph, 2, x2, y2, 3);
+    EN_addcurve(ph, curve3);
+    EN_setcurve(ph, 3, x3, y3, 1);
+
+    // Assign Curve3 to pump's head curve
+    EN_getcurveindex(ph, curve3, &curveIdx);
+    EN_setheadcurveindex(ph, pumpIdx, curveIdx);
+
+    // Delete Curve2
+    EN_getcurveindex(ph, curve2, &curveIdx);
+    EN_deletecurve(ph, curveIdx);
+
+    // Check that pump's head curve is still Curve3
+    EN_getheadcurveindex(ph, pumpIdx, &curveIdx);
+    EN_getcurveid(ph, curveIdx, curveID);
+    BOOST_REQUIRE(strcmp(curveID, curve3) == 0);
+
+    // And that it contains the correct data
+    double x, y;
+    EN_getcurvevalue(ph, curveIdx, 1, &x, &y);
+    BOOST_REQUIRE(x == x3[0]);
+    BOOST_REQUIRE(y == y3[0]);
+
+    EN_close(ph);
+    EN_deleteproject(&ph);
+}
+
+BOOST_AUTO_TEST_SUITE_END()