def test_vars_are_bound_or_free_or_constants(self):
t = '''
((or (a ?method_var str_const1) (b ?method_var str_const2)) (c ?method_var ?out_var) (d ?out_var 123.0))
'''
actual = logical_expression.build(s_expression.parse(t))
actual = logical_expression.convert_to_dnf(actual)
bound_variables = ['?method_var']
logical_expression.classify_variables(bound_variables=bound_variables, root_node=actual)
# bound_variables argument isn't mutable
self.assertEqual(['?method_var'], bound_variables)
a_node = list(self.world_state_facts(actual, 'a'))[0]
b_node = list(self.world_state_facts(actual, 'b'))[0]
c_node = list(self.world_state_facts(actual, 'c'))[0]
d_node = list(self.world_state_facts(actual, 'd'))[0]
self.assertEqual(set(['?method_var']), set(a_node.bound_variables))
self.assertEqual(set(['str_const1']), set(a_node.constants))
self.assertEqual(set(['?method_var']), set(b_node.bound_variables))
self.assertEqual(set(['str_const2']), set(b_node.constants))
self.assertEqual(set(['?method_var']), set(c_node.bound_variables))
self.assertEqual(set(), set(c_node.constants))
self.assertEqual(set(['?out_var']), set(d_node.bound_variables))
self.assertEqual(set([123.0]), set(d_node.constants))
def build_branch(name, precondition_expr, tasklist_expr, method_node):
branch = node_type.Branch()
branch.name = lex_token.fixup_name(name)
if match_child(precondition_expr, 0, s_expression.Symbol, lex_token.PERMUTE):
permute_atom_expr = assert_child(precondition_expr, 1, s_expression.List)
branch.permute_atom = build_permute_atom(permute_atom_expr)
precondition_expr.children = precondition_expr.children[2:]
precondition_node = logical_expression.build(precondition_expr)
precondition_node = logical_expression.convert_to_dnf(precondition_node)
logical_expression.classify_variables(method_node.variables, precondition_node)
task_list_nodes = build_task_list(tasklist_expr)
branch.children = [precondition_node] + task_list_nodes
return branch
def build_branch(name, precondition_expr, tasklist_expr, method_node):
branch = node_type.Branch()
branch.name = lex_token.fixup_name(name)
if match_child(precondition_expr, 0, s_expression.Symbol,
lex_token.PERMUTE):
permute_atom_expr = assert_child(precondition_expr, 1,
s_expression.List)
branch.permute_atom = build_permute_atom(permute_atom_expr)
precondition_expr.children = precondition_expr.children[2:]
precondition_node = logical_expression.build(precondition_expr)
precondition_node = logical_expression.convert_to_dnf(precondition_node)
logical_expression.classify_variables(method_node.variables,
precondition_node)
task_list_nodes = build_task_list(tasklist_expr)
branch.children = [precondition_node] + task_list_nodes
return branch
