1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
|
/*************************************************************************
* Copyright (C) 2018-2022 Blue Brain Project
*
* This file is part of NMODL distributed under the terms of the GNU
* Lesser General Public License. See top-level LICENSE file for details.
*************************************************************************/
#include <catch2/catch_test_macros.hpp>
#include "ast/program.hpp"
#include "parser/nmodl_driver.hpp"
#include "test/unit/utils/test_utils.hpp"
#include "visitors/after_cvode_to_cnexp_visitor.hpp"
#include "visitors/checkparent_visitor.hpp"
#include "visitors/symtab_visitor.hpp"
#include "visitors/visitor_utils.hpp"
using namespace nmodl;
using namespace visitor;
using namespace test;
using namespace test_utils;
using nmodl::parser::NmodlDriver;
//=============================================================================
// AfterCVodeToCnexp visitor tests
//=============================================================================
std::string run_after_cvode_to_cnexp_visitor(const std::string& text) {
NmodlDriver driver;
const auto& ast = driver.parse_string(text);
SymtabVisitor().visit_program(*ast);
AfterCVodeToCnexpVisitor().visit_program(*ast);
// check that, after visitor rearrangement, parents are still up-to-date
CheckParentVisitor().check_ast(*ast);
return to_nmodl(ast);
}
SCENARIO("AfterCVodeToCnexpVisitor changes after_cvode solver method to cnexp") {
GIVEN("Breakpoint block with after_cvode method") {
std::string nmodl_text = R"(
BREAKPOINT {
SOLVE states METHOD after_cvode
}
)";
std::string output_nmodl = R"(
BREAKPOINT {
SOLVE states METHOD cnexp
}
)";
THEN("AfterCVodeToCnexp visitor replaces after_cvode solver with cnexp") {
std::string input = reindent_text(nmodl_text);
auto expected_result = reindent_text(output_nmodl);
auto result = run_after_cvode_to_cnexp_visitor(input);
REQUIRE(result == expected_result);
}
}
}
|