def test_take_action_gives_correct_next_states_with_lambda_productions(self):
# state.take_action() doesn't read or change these objects, it just passes them through, so
# we'll use some sentinels to be sure of that.
valid_actions = object()
action_indices = object()
state = GrammarState(['t', '<s,d>'], {}, valid_actions, action_indices, is_nonterminal)
next_state = state.take_action('<s,d> -> [lambda x, d]')
expected_next_state = GrammarState(['t', 'd'],
{('s', 'x'): ['d']},
valid_actions,
action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action('d -> [<s,r>, d]')
expected_next_state = GrammarState(['t', 'd', '<s,r>'],
{('s', 'x'): ['d', '<s,r>']},
valid_actions,
action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action('<s,r> -> [lambda y, r]')
expected_next_state = GrammarState(['t', 'd', 'r'],
{('s', 'x'): ['d', 'r'], ('s', 'y'): ['r']},
valid_actions,
action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action('r -> identity')
expected_next_state = GrammarState(['t', 'd'],
{('s', 'x'): ['d']},
valid_actions,
action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action('d -> x')
expected_next_state = GrammarState(['t'],
{},
valid_actions,
action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
def test_take_action_gives_correct_next_states_with_non_lambda_productions(self):
# state.take_action() doesn't read or change these objects, it just passes them through, so
# we'll use some sentinels to be sure of that.
valid_actions = object()
action_indices = object()
state = GrammarState(['s'], {}, valid_actions, action_indices, is_nonterminal)
next_state = state.take_action('s -> [t, r]')
expected_next_state = GrammarState(['r', 't'], {}, valid_actions, action_indices, is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = GrammarState(['r', 't'], {}, valid_actions, action_indices, is_nonterminal)
next_state = state.take_action('t -> identity')
expected_next_state = GrammarState(['r'], {}, valid_actions, action_indices, is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
def test_take_action_gives_correct_next_states_with_lambda_productions(
self):
# state.take_action() doesn't read or change these objects, it just passes them through, so
# we'll use some sentinels to be sure of that.
valid_actions = object()
action_indices = object()
state = GrammarState([u't', u'<s,d>'], {}, valid_actions,
action_indices, is_nonterminal)
next_state = state.take_action(u'<s,d> -> [lambda x, d]')
expected_next_state = GrammarState([u't', u'd'],
{(u's', u'x'): [u'd']},
valid_actions, action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action(u'd -> [<s,r>, d]')
expected_next_state = GrammarState([u't', u'd', u'<s,r>'],
{(u's', u'x'): [u'd', u'<s,r>']},
valid_actions, action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action(u'<s,r> -> [lambda y, r]')
expected_next_state = GrammarState([u't', u'd', u'r'], {
(u's', u'x'): [u'd', u'r'],
(u's', u'y'): [u'r']
}, valid_actions, action_indices, is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action(u'r -> identity')
expected_next_state = GrammarState([u't', u'd'],
{(u's', u'x'): [u'd']},
valid_actions, action_indices,
is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = expected_next_state
next_state = state.take_action(u'd -> x')
expected_next_state = GrammarState([u't'], {}, valid_actions,
action_indices, is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
def test_take_action_gives_correct_next_states_with_non_lambda_productions(
self):
# state.take_action() doesn't read or change these objects, it just passes them through, so
# we'll use some sentinels to be sure of that.
valid_actions = object()
context_actions = object()
state = GrammarState(['s'], {}, valid_actions, context_actions,
is_nonterminal)
next_state = state.take_action('s -> [t, r]')
expected_next_state = GrammarState(['r', 't'], {}, valid_actions,
context_actions, is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
state = GrammarState(['r', 't'], {}, valid_actions, context_actions,
is_nonterminal)
next_state = state.take_action('t -> identity')
expected_next_state = GrammarState(['r'], {}, valid_actions,
context_actions, is_nonterminal)
assert next_state.__dict__ == expected_next_state.__dict__
def test_take_action_crashes_with_mismatched_types(self):
with pytest.raises(AssertionError):
state = GrammarState(['s'], {}, {}, {}, is_nonterminal)
state.take_action('t -> identity')
def test_take_action_crashes_with_mismatched_types(self):
with pytest.raises(AssertionError):
state = GrammarState(['s'], {}, {}, {}, is_nonterminal)
state.take_action('t -> identity')