85 lines
2.7 KiB
C++
85 lines
2.7 KiB
C++
#ifdef WITH_MODULE
|
|
import argparse;
|
|
#else
|
|
#include <argparse/argparse.hpp>
|
|
#endif
|
|
#include <doctest.hpp>
|
|
#include <test_utility.hpp>
|
|
|
|
using doctest::test_suite;
|
|
|
|
TEST_CASE("Parse vector of arguments" * test_suite("vector")) {
|
|
argparse::ArgumentParser program("test");
|
|
program.add_argument("input").nargs(2);
|
|
|
|
program.parse_args({"test", "rocket.mesh", "thrust_profile.csv"});
|
|
|
|
auto inputs = program.get<std::vector<std::string>>("input");
|
|
REQUIRE(inputs.size() == 2);
|
|
REQUIRE(inputs[0] == "rocket.mesh");
|
|
REQUIRE(inputs[1] == "thrust_profile.csv");
|
|
}
|
|
|
|
TEST_CASE("Parse list of arguments" * test_suite("vector")) {
|
|
argparse::ArgumentParser program("test");
|
|
program.add_argument("input").nargs(2);
|
|
|
|
program.parse_args({"test", "rocket.mesh", "thrust_profile.csv"});
|
|
|
|
auto inputs = program.get<std::list<std::string>>("input");
|
|
REQUIRE(inputs.size() == 2);
|
|
REQUIRE(testutility::get_from_list(inputs, 0) == "rocket.mesh");
|
|
REQUIRE(testutility::get_from_list(inputs, 1) == "thrust_profile.csv");
|
|
}
|
|
|
|
TEST_CASE("Parse list of arguments with default values" *
|
|
test_suite("vector")) {
|
|
argparse::ArgumentParser program("test");
|
|
program.add_argument("--input")
|
|
.default_value(std::list<int>{1, 2, 3, 4, 5})
|
|
.nargs(5);
|
|
|
|
program.parse_args({"test"});
|
|
|
|
auto inputs = program.get<std::list<int>>("--input");
|
|
REQUIRE(inputs.size() == 5);
|
|
REQUIRE(testutility::get_from_list(inputs, 0) == 1);
|
|
REQUIRE(testutility::get_from_list(inputs, 1) == 2);
|
|
REQUIRE(testutility::get_from_list(inputs, 2) == 3);
|
|
REQUIRE(testutility::get_from_list(inputs, 3) == 4);
|
|
REQUIRE(testutility::get_from_list(inputs, 4) == 5);
|
|
REQUIRE(program["--input"] == std::list<int>{1, 2, 3, 4, 5});
|
|
}
|
|
|
|
TEST_CASE("Parse list of arguments and save in an object" *
|
|
test_suite("vector")) {
|
|
|
|
struct ConfigManager {
|
|
std::vector<std::string> files;
|
|
void add_file(const std::string &file) { files.push_back(file); }
|
|
};
|
|
|
|
ConfigManager config_manager;
|
|
|
|
argparse::ArgumentParser program("test");
|
|
program.add_argument("--input_files")
|
|
.nargs(2)
|
|
.action([&](const std::string &value) {
|
|
config_manager.add_file(value);
|
|
return value;
|
|
});
|
|
|
|
program.parse_args({"test", "--input_files", "config.xml", "system.json"});
|
|
|
|
auto file_args = program.get<std::vector<std::string>>("--input_files");
|
|
REQUIRE(file_args.size() == 2);
|
|
REQUIRE(file_args[0] == "config.xml");
|
|
REQUIRE(file_args[1] == "system.json");
|
|
|
|
REQUIRE(config_manager.files.size() == 2);
|
|
REQUIRE(config_manager.files[0] == "config.xml");
|
|
REQUIRE(config_manager.files[1] == "system.json");
|
|
REQUIRE(program["--input_files"] ==
|
|
std::vector<std::string>{"config.xml", "system.json"});
|
|
}
|