def test_parse_predicates(self): parser = PDDL_Parser() parser.predicates = {} parser.parse_predicates( [['untyped_pred', '?v1', '?v2', '?v3'], [ 'typed_pred', '?v1', '-', 'type1', '?v2', '-', 'type1', '?v3', '-', 'object' ], ['shared_type_pred', '?v1', '?v2', '-', 'type1', '?v3']]) self.assertEqual( parser.predicates, { 'untyped_pred': { '?v1': 'object', '?v2': 'object', '?v3': 'object' }, 'typed_pred': { '?v1': 'type1', '?v2': 'type1', '?v3': 'object' }, 'shared_type_pred': { '?v1': 'type1', '?v2': 'type1', '?v3': 'object' } })