def test_concat_any_number_same(fa, fb, fc):
p: Node = fa + AnyNumber(fb)
exp1 = p.copy()
g = ast_to_graph(exp1)
edges = list(
sorted([("I", "a"), ("a", "b"), ("a", "T"), ("b", "b"), ("b", "T")]))
assert graph_edges(g) == edges
exp2 = (p + fc).copy()
g = ast_to_graph(exp2)
edges = list(
sorted([("I", "a"), ("a", "b"), ("a", "c"), ("b", "b"), ("b", "c"),
("c", "T")]))
assert graph_edges(g) == edges
exp3 = (exp2 + p).copy()
g = ast_to_graph(exp3.copy())
edges = list(
sorted([
("I", "a"),
("a", "c"),
("a", "b"),
("b", "b"),
("b", "c"),
("c", "a"),
("a", "T"),
("a", "b"),
("b", "b"),
("b", "T"),
]))
assert graph_edges(g) == edges
def test_concat_any_number():
a1 = Final(Eq("a1"))
a2 = Final(Eq("a2"))
b1 = Final(Eq("b1"))
b2 = Final(Eq("b2"))
c = Final(Eq("c"))
p1: Node = a1 + AnyNumber(b1)
p2: Node = a2 + AnyNumber(b2)
exp = p1 + c + p2
g = ast_to_graph(exp.copy())
edges = list(
sorted([
("I", "a1"),
("a1", "c"),
("a1", "b1"),
("b1", "b1"),
("b1", "c"),
("c", "a2"),
("a2", "T"),
("a2", "b2"),
("b2", "b2"),
("b2", "T"),
]))
assert graph_edges(g) == edges
def test_exp_1(fa, fb, fc):
g = ast_to_graph(((fa + fb) | fc) * slice(0, None))
assert {*g.successors(_Initial())} == {fa, fc, _Terminal()}
assert {*g.successors(fa)} == {fb}
assert {*g.successors(fb)} == {fa, fc, _Terminal()}
assert {*g.successors(fc)} == {fa, fc, _Terminal()}
def test_any_number_capture_2(fa, fb):
cap = fa["foo"]
g = ast_to_graph(AnyNumber(cap))
assert g.edges[_Initial(), fa].get("start_captures", []) == [cap]
assert g.edges[_Initial(), fa].get("stop_captures", []) == []
assert g.edges[fa, fa].get("start_captures", []) == [cap]
assert g.edges[fa, fa].get("stop_captures", []) == [cap]
assert g.edges[fa, _Terminal()].get("start_captures", []) == []
assert g.edges[fa, _Terminal()].get("stop_captures", []) == [cap]
assert g.edges[_Initial(), _Terminal()].get("start_captures", []) == []
assert g.edges[_Initial(), _Terminal()].get("stop_captures", []) == []
def test_capture_on_capture(fa, fb):
cap1 = fa["foo"]
cap2 = fb["bar"]
exp = cap1 + cap2
g = ast_to_graph(exp)
assert g.edges[_Initial(), fa].get("start_captures", []) == [cap1]
assert g.edges[_Initial(), fa].get("stop_captures", []) == []
assert g.edges[fa, fb].get("start_captures", []) == [cap2]
assert g.edges[fa, fb].get("stop_captures", []) == [cap1]
assert g.edges[fb, _Terminal()].get("start_captures", []) == []
assert g.edges[fb, _Terminal()].get("stop_captures", []) == [cap2]
def test_any_number_capture_3(fa, fb):
cap1 = AnyNumber(fa)["foo"]
cap2 = fb["bar"]
g = ast_to_graph(cap1 + cap2)
edges = list(
sorted([
("I", "a"),
("I", "b"),
("a", "a"),
("a", "b"),
("b", "T"),
]))
assert graph_edges(g) == edges
assert g.edges[_Initial(), fa].get("start_captures", []) == [cap1]
assert g.edges[_Initial(), fa].get("stop_captures", []) == []
assert g.edges[_Initial(), fb].get("start_captures", []) == [cap2]
assert g.edges[_Initial(), fb].get("stop_captures", []) == []
assert g.edges[fa, fa].get("start_captures", []) == []
assert g.edges[fa, fa].get("stop_captures", []) == []
assert g.edges[fa, fb].get("start_captures", []) == [cap2]
assert g.edges[fa, fb].get("stop_captures", []) == [cap1]
assert g.edges[fb, _Terminal()].get("start_captures", []) == []
assert g.edges[fb, _Terminal()].get("stop_captures", []) == [cap2]
def test_exp_2(fa, fb):
g = ast_to_graph(AnyNumber(AnyNumber(fa) | AnyNumber(fb)))
assert {*g.successors(_Initial())} == {fa, fb, _Terminal()}
assert {*g.successors(fa)} == {fa, fb, _Terminal()}
assert {*g.successors(fb)} == {fa, fb, _Terminal()}
def test_capture(fa):
capture = Capture(statement=fa, name="foo")
g = ast_to_graph(capture)
assert g.get_edge_data(_Initial(), fa) == {"start_captures": [capture]}
assert g.get_edge_data(fa, _Terminal()) == {"stop_captures": [capture]}
def test_any_number(fa):
g = ast_to_graph(AnyNumber(fa))
assert {*g.successors(_Initial())} == {fa, _Terminal()}
assert {*g.successors(fa)} == {fa, _Terminal()}
def test_maybe(fa):
g = ast_to_graph(Maybe(fa))
assert {*g.successors(_Initial())} == {fa, _Terminal()}
assert {*g.successors(fa)} == {_Terminal()}
def test_alternate(fa, fb):
g = ast_to_graph(fa | fb)
assert {*g.successors(_Initial())} == {fa, fb}
assert {*g.successors(fa)} == {_Terminal()}
assert {*g.successors(fb)} == {_Terminal()}
def test_concatenate(fa, fb):
g = ast_to_graph(fa + fb)
assert [*g.successors(_Initial())] == [fa]
assert [*g.successors(fa)] == [fb]
assert [*g.successors(fb)] == [_Terminal()]
def test_final(fa):
g = ast_to_graph(fa)
assert [*g.successors(_Initial())] == [fa]
assert [*g.successors(fa)] == [_Terminal()]