Adds new test grammar and lr1 states

Author: Maxcode123

src/syntactes/tests/data.py

@@ -1,18 +1,22 @@
 from syntactes import Grammar, Rule, Token
 from syntactes._action import Action
-from syntactes._item import LR0Item
-from syntactes._state import LR0State
+from syntactes._item import LR0Item, LR1Item
+from syntactes._state import LR0State, LR1State
 from syntactes.parsing_table import Entry, LR0ParsingTable, SLRParsingTable
 
 EOF = Token.eof()
 S = Token("S", False)
 E = Token("E", False)
 T = Token("T", False)
+L = Token("L", False)
+C = Token("C", False)
 x = Token("x", True)
 PLUS = Token("+", True)
+LPAREN = Token("(", True)
+RPAREN = Token(")", True)
 
 tokens_1 = {EOF, S, E, T, x, PLUS}
-
+tokens_2 = {EOF, S, L, C, LPAREN, RPAREN}
 
 # 1. S -> E $
 # 2. E -> T + E
@@ -27,6 +31,21 @@
 
 grammar_1 = Grammar(rule_1_1, rules_1, tokens_1)
 
+# 1. S -> L $
+# 2. L -> L C
+# 3. L -> C
+# 4. C -> LPAREN C RPAREN
+# 5. C -> LPAREN RPAREN
+rule_1_2 = Rule(0, S, L, EOF)
+rule_2_2 = Rule(1, L, L, C)
+rule_3_2 = Rule(2, L, C)
+rule_4_2 = Rule(3, C, LPAREN, C, RPAREN)
+rule_5_2 = Rule(4, C, LPAREN, RPAREN)
+
+rules_2 = (rule_1_2, rule_2_2, rule_3_2, rule_4_2, rule_5_2)
+
+grammar_2 = Grammar(rule_1_2, rules_2, tokens_2)
+
 
 def lr0_state_1():
     item_1 = LR0Item(grammar_1.starting_rule, 0)  # S -> . E $
@@ -115,3 +134,119 @@ def slr_parsing_table():
     table.add_entry(Entry(lr0_state_5(), EOF, Action.reduce(rule_4_1)))
     table.add_entry(Entry(lr0_state_6(), EOF, Action.reduce(rule_2_1)))
     return table
+
+
+def lr1_state_1():
+    item_1 = LR1Item(rule_1_2, 0, EOF)  # S -> . L $, $
+    item_2 = LR1Item(rule_2_2, 0, EOF)  # L -> . L C, $
+    item_3 = LR1Item(rule_2_2, 0, LPAREN)  # L -> . L C, (
+    item_4 = LR1Item(rule_3_2, 0, EOF)  # L -> . C, $
+    item_5 = LR1Item(rule_3_2, 0, LPAREN)  # L -> . C, (
+    item_6 = LR1Item(rule_4_2, 0, EOF)  # C -> . ( C ), $
+    item_7 = LR1Item(rule_4_2, 0, LPAREN)  # C -> . ( C ), (
+    item_8 = LR1Item(rule_5_2, 0, EOF)  # C -> . (), $
+    item_9 = LR1Item(rule_5_2, 0, LPAREN)  # C -> . (), (
+    state = LR1State.from_items(
+        {item_1, item_2, item_3, item_4, item_5, item_6, item_7, item_8, item_9}
+    )
+    state.set_number(1)
+    return state
+
+
+def lr1_state_2():
+    item_1 = LR1Item(rule_1_2, 1, EOF)  # S -> L . $, $
+    item_2 = LR1Item(rule_2_2, 1, EOF)  # L -> L . C, $
+    item_3 = LR1Item(rule_2_2, 1, LPAREN)  # L -> L . C, (
+    item_4 = LR1Item(rule_4_2, 0, EOF)  # C -> . ( C ), $
+    item_5 = LR1Item(rule_4_2, 0, LPAREN)  # C -> . ( C ), (
+    item_6 = LR1Item(rule_5_2, 0, EOF)  # C -> . ( ), $
+    item_7 = LR1Item(rule_5_2, 0, LPAREN)  # C -> . ( ), (
+    state = LR1State.from_items(
+        {item_1, item_2, item_3, item_4, item_5, item_6, item_7}
+    )
+    state.set_number(2)
+    state.set_final()
+    return state
+
+
+def lr1_state_3():
+    item_1 = LR1Item(rule_3_2, 1, EOF)  # L -> C ., $
+    item_2 = LR1Item(rule_3_2, 1, LPAREN)  # L -> C ., (
+    state = LR1State.from_items({item_1, item_2})
+    state.set_number(3)
+    return state
+
+
+def lr1_state_4():
+    item_1 = LR1Item(rule_4_2, 0, RPAREN)  # C -> . ( C ), )
+    item_2 = LR1Item(rule_4_2, 1, EOF)  # C -> ( . C ), $
+    item_3 = LR1Item(rule_4_2, 1, LPAREN)  # C -> ( . C ), (
+    item_4 = LR1Item(rule_5_2, 0, RPAREN)  # C -> . ( ), )
+    item_5 = LR1Item(rule_5_2, 1, EOF)  # C -> ( . ), $
+    item_6 = LR1Item(rule_5_2, 1, LPAREN)  # C -> ( . ), (
+    state = LR1State.from_items({item_1, item_2, item_3, item_4, item_5, item_6})
+    state.set_number(4)
+    return state
+
+
+def lr1_state_5():
+    item_1 = LR1Item(rule_2_2, 2, EOF)  # L -> L C ., $
+    item_2 = LR1Item(rule_2_2, 2, LPAREN)  # L -> L C ., (
+    state = LR1State.from_items({item_1, item_2})
+    state.set_number(5)
+    return state
+
+
+def lr1_state_6():
+    item_1 = LR1Item(rule_4_2, 2, EOF)  # C -> ( C . ), $
+    item_2 = LR1Item(rule_4_2, 2, LPAREN)  # C -> ( C . ), (
+    state = LR1State.from_items({item_1, item_2})
+    state.set_number(6)
+    return state
+
+
+def lr1_state_7():
+    item_1 = LR1Item(rule_4_2, 0, RPAREN)  # C -> . ( C ), )
+    item_2 = LR1Item(rule_4_2, 1, RPAREN)  # C -> ( . C ), )
+    item_3 = LR1Item(rule_5_2, 0, RPAREN)  # C -> . ( ), )
+    item_4 = LR1Item(rule_5_2, 1, RPAREN)  # C -> ( . ), )
+    state = LR1State.from_items({item_1, item_2, item_3, item_4})
+    state.set_number(7)
+    return state
+
+
+def lr1_state_8():
+    item_1 = LR1Item(rule_5_2, 2, EOF)  # C -> ( ) ., $
+    item_2 = LR1Item(rule_5_2, 2, LPAREN)  # C -> ( ) ., (
+    state = LR1State.from_items({item_1, item_2})
+    state.set_number(8)
+    return state
+
+
+def lr1_state_9():
+    item_1 = LR1Item(rule_4_2, 3, EOF)  # C -> ( C ) ., $
+    item_2 = LR1Item(rule_4_2, 3, LPAREN)  # C -> ( C ) ., (
+    state = LR1State.from_items({item_1, item_2})
+    state.set_number(9)
+    return state
+
+
+def lr1_state_10():
+    item_1 = LR1Item(rule_4_2, 2, RPAREN)  # C -> ( C . ), )
+    state = LR1State.from_items({item_1})
+    state.set_number(10)
+    return state
+
+
+def lr1_state_11():
+    item_1 = LR1Item(rule_5_2, 2, RPAREN)  # C -> ( ) ., )
+    state = LR1State.from_items({item_1})
+    state.set_number(11)
+    return state
+
+
+def lr1_state_12():
+    item_1 = LR1Item(rule_4_2, 3, RPAREN)  # C -> ( C ) ., )
+    state = LR1State.from_items({item_1})
+    state.set_number(12)
+    return state