tree-sitter/spec/compiler/build_tables/perform_spec.cpp
2014-01-23 13:10:37 -08:00

110 lines
3.1 KiB
C++

#include "spec_helper.h"
#include <functional>
#include "build_tables/perform.h"
using build_tables::perform;
using namespace rules;
typedef unordered_set<ParseAction> parse_actions;
typedef unordered_set<LexAction> lex_actions;
static unordered_set<string> keys(const unordered_map<string, parse_actions> &map) {
unordered_set<string> result;
for (auto pair : map) {
result.insert(pair.first);
}
return result;
}
static unordered_set<CharMatch> keys(const unordered_map<CharMatch, lex_actions> &map) {
unordered_set<CharMatch> result;
for (auto pair : map) {
result.insert(pair.first);
}
return result;
}
START_TEST
describe("building parse and lex tables", []() {
Grammar grammar({
{ "expression", choice({
seq({
sym("term"),
sym("plus"),
sym("term") }),
sym("term") }) },
{ "term", choice({
sym("variable"),
sym("number"),
seq({
sym("left-paren"),
sym("expression"),
sym("right-paren")
}) }) }
});
Grammar lex_grammar({
{ "plus", str("+") },
{ "variable", pattern("\\w+") },
{ "number", pattern("\\d+") },
{ "left-paren", str("(") },
{ "right-paren", str(")") }
});
ParseTable table;
LexTable lex_table;
before_each([&]() {
pair<ParseTable, LexTable> tables = perform(grammar, lex_grammar);
table = tables.first;
lex_table = tables.second;
});
function<ParseState(size_t)> parse_state = [&](size_t index) {
return table.states[index];
};
function<LexState(size_t)> lex_state = [&](size_t parse_state_index) {
long index = table.states[parse_state_index].lex_state_index;
return lex_table.states[index];
};
it("has the right starting state", [&]() {
AssertThat(keys(parse_state(0).actions), Equals(unordered_set<string>({
"expression",
"term",
"number",
"variable",
"left-paren",
})));
AssertThat(keys(lex_state(0).actions), Equals(unordered_set<CharMatch>({
CharMatchSpecific('('),
CharMatchClass(CharClassDigit),
CharMatchClass(CharClassWord),
})));
AssertThat(lex_state(0).expected_inputs(), Equals(unordered_set<CharMatch>({
CharMatchSpecific('('),
CharMatchClass(CharClassDigit),
CharMatchClass(CharClassWord),
})));
});
it("accepts when the start symbol is reduced", [&]() {
AssertThat(parse_state(1).actions, Equals(unordered_map<string, parse_actions>({
{ ParseTable::END_OF_INPUT, parse_actions({ ParseAction::Accept() }) }
})));
});
it("has the right next states", [&]() {
AssertThat(parse_state(2).actions, Equals(unordered_map<string, parse_actions>({
{ "plus", parse_actions({ ParseAction::Shift(3) }) },
{ "__END__", parse_actions({ ParseAction::Reduce("expression", 1) }) },
})));
});
});
END_TEST