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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
/*************************************************************************
* 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/nmodl_constructs.hpp"
#include "visitors/local_to_assigned_visitor.hpp"
#include "visitors/nmodl_visitor.hpp"
#include "visitors/perf_visitor.hpp"
#include "visitors/symtab_visitor.hpp"
using namespace nmodl;
using namespace visitor;
using namespace test_utils;
using nmodl::parser::NmodlDriver;
using symtab::syminfo::NmodlType;
//=============================================================================
// GlobalToRange visitor tests
//=============================================================================
std::shared_ptr<ast::Program> run_local_to_assigned_visitor(const std::string& text) {
std::map<std::string, std::string> rval;
NmodlDriver driver;
auto ast = driver.parse_string(text);
SymtabVisitor().visit_program(*ast);
PerfVisitor().visit_program(*ast);
LocalToAssignedVisitor().visit_program(*ast);
SymtabVisitor().visit_program(*ast);
return ast;
}
SCENARIO("LOCAL to ASSIGNED variable transformer", "[visitor][localtoassigned]") {
GIVEN("mod file with LOCAL variables that are written") {
std::string input_nmodl = R"(
NEURON {
SUFFIX test
}
LOCAL x, y, z
INITIAL {
x = 1
}
BREAKPOINT {
z = 2
}
)";
auto ast = run_local_to_assigned_visitor(input_nmodl);
auto symtab = ast->get_symbol_table();
THEN("LOCAL variables that are written are turned to ASSIGNED") {
/// check for all ASSIGNED variables : old ones + newly converted ones
auto vars = symtab->get_variables_with_properties(NmodlType::assigned_definition);
REQUIRE(vars.size() == 2);
/// x and z should be converted from LOCAL to ASSIGNED
auto x = symtab->lookup("x");
REQUIRE(x != nullptr);
REQUIRE(x->has_any_property(NmodlType::assigned_definition) == true);
REQUIRE(x->has_any_property(NmodlType::local_var) == false);
auto z = symtab->lookup("z");
REQUIRE(z != nullptr);
REQUIRE(z->has_any_property(NmodlType::assigned_definition) == true);
REQUIRE(z->has_any_property(NmodlType::local_var) == false);
}
THEN("LOCAL variables that are read only remain LOCAL") {
auto vars = symtab->get_variables_with_properties(NmodlType::local_var);
REQUIRE(vars.size() == 1);
REQUIRE(vars[0]->get_name() == "y");
}
}
}
|
