rename_function_arguments.cpp

Go to the documentation of this file.

#include <catch2/catch_test_macros.hpp>
#include <catch2/matchers/catch_matchers_string.hpp>
 
#include "ast/program.hpp"
#include "parser/nmodl_driver.hpp"
#include "visitors/rename_function_arguments.hpp"
#include "visitors/visitor_utils.hpp"
 
using namespace nmodl;
using namespace visitor;
 
using Catch::Matchers::ContainsSubstring;
using nmodl::parser::NmodlDriver;
 
 
TEST_CASE("Rename function arguments") {
    std::string mod_code = R"(
NEURON {
    RANGE tau
}
 
FUNCTION foo(a, v, cai) {
    exp(tau + a + v*cai)
}
 
PROCEDURE bar(tau, a, b, d) {
    foo(tau, a*b, d)
}
)";
 
    NmodlDriver driver;
    const auto& ast = driver.parse_string(mod_code);
 
    auto visitor = nmodl::visitor::RenameFunctionArgumentsVisitor();
    visitor.visit_program(*ast);
 
    auto transformed_mod = nmodl::to_nmodl(*ast);
 
    SECTION("FUNCTION") {
        std::string expected = R"(
FUNCTION foo(_la, _lv, _lcai) {
    exp(tau+_la+_lv*_lcai)
}
)";
 
        REQUIRE_THAT(transformed_mod, ContainsSubstring(expected));
    }
 
    SECTION("PROCEDURE") {
        std::string expected = R"(
PROCEDURE bar(_ltau, _la, _lb, _ld) {
    foo(_ltau, _la*_lb, _ld)
}
)";
 
        REQUIRE_THAT(transformed_mod, ContainsSubstring(expected));
    }
}