initial_block.cpp

Go to the documentation of this file.

/*
 * Copyright 2025 EPFL.
 * See the top-level LICENSE file for details.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#include "ast/program.hpp"
#include "parser/nmodl_driver.hpp"
#include "utils/test_utils.hpp"
#include "visitors/initial_block_visitor.hpp"
#include "visitors/nmodl_visitor.hpp"
#include "visitors/symtab_visitor.hpp"
#include "visitors/visitor_utils.hpp"
#include "visitors/json_visitor.hpp"
 
#include <catch2/catch_test_macros.hpp>
#include <catch2/matchers/catch_matchers_string.hpp>
 
using namespace nmodl;
using nmodl::test_utils::reindent_text;
 
using Catch::Matchers::ContainsSubstring;  // ContainsSubstring in newer Catch2
 
//=============================================================================
// MergeInitialBlocks visitor tests
//=============================================================================
 
auto generate_mod_after_merge_initial_blocks_visitor(std::string const& text) {
    parser::NmodlDriver driver{};
    auto ast = driver.parse_string(text);
    visitor::MergeInitialBlocksVisitor{}.visit_program(*ast);
    return to_nmodl(*ast);
}
 
SCENARIO("Check multiple INITIAL blocks are handled properly", "[visitor][merge_initial_blocks]") {
    GIVEN("A mod file with multiple INITIAL blocks") {
        const auto nmodl_text_before = R"(
            NEURON {
                SUFFIX InitialBlockTest
                RANGE foo, bar
            }
            INITIAL {
                foo = 1
            }
            INITIAL {
                bar = 2
            }
        )";
        const auto nmodl_text_after = R"(
            NEURON {
              SUFFIX InitialBlockTest
              RANGE foo, bar
            }
            INITIAL {
              foo = 1
              bar = 2
            }
        )";
        parser::NmodlDriver driver{};
        auto ast_expected = driver.parse_string(nmodl_text_after);
        const auto program_expected = to_nmodl(ast_expected);
        const auto program_actual = generate_mod_after_merge_initial_blocks_visitor(
            nmodl_text_before);
        THEN("expected and actual should be identical at the level of the AST") {
            // TODO the AST class lacks an overload for `operator==` so here we compare it at the
            // string level
            REQUIRE(reindent_text(program_actual) == reindent_text(program_expected));
        }
    }
    GIVEN("A mod file with an INITIAL block only inside of a NET_RECEIVE block") {
        const auto nmodl_text_before = R"(
            NEURON {
                SUFFIX test
                RANGE foo, bar
            }
 
            NET_RECEIVE (w) {
                INITIAL {
                    foo = 1
                }
            }
        )";
        const auto program_actual = generate_mod_after_merge_initial_blocks_visitor(
            nmodl_text_before);
        THEN("leave the mod file as-is") {
            REQUIRE(reindent_text(program_actual) == reindent_text(nmodl_text_before));
        }
    }
    GIVEN("A mod file with an INITIAL block, and one inside of a NET_RECEIVE block") {
        // Note that the visitor actually modifies the AST (since there is > 1 INITIAL block in the
        // entire file: one top-level, and one in NET_RECEIVE). If we place the top-level INITIAL
        // block before NET_RECEIVE in the below, the top-level INITIAL block will be deleted and
        // appended. However, since the position of the INITIAL block in the mod file has no impact
        // on the semantics, the visitor works as expected
        const auto nmodl_text_before = R"(
            NEURON {
                SUFFIX test
                RANGE foo, bar
            }
 
            NET_RECEIVE (w) {
                INITIAL {
                    foo = 1
                }
            }
 
            INITIAL {
                bar = 2
            }
        )";
        const auto program_actual = generate_mod_after_merge_initial_blocks_visitor(
            nmodl_text_before);
        THEN("leave the mod file as-is") {
            REQUIRE(reindent_text(program_actual) == reindent_text(nmodl_text_before));
        }
    }
}