class TestEmbeddings (unittest.TestCase):
def setUp(self):
#
# db, store
#
db_url = 'sqlite:///foo.db'
# setup compiler + environment
self.db = LogicDB(db_url, echo=False)
self.parser = PrologParser(self.db)
self.rt = PrologRuntime(self.db)
# self.rt.set_trace(True)
self.db.clear_module(UNITTEST_MODULE)
def tearDown(self):
self.db.close()
#@unittest.skip("temporarily disabled")
def test_custom_builtins(self):
global recorded_moves
self.parser.compile_file('samples/hanoi2.pl', UNITTEST_MODULE)
clause = self.parser.parse_line_clause_body('move(3,left,right,center)')
logging.debug('clause: %s' % clause)
# register our custom builtin
recorded_moves = []
self.rt.register_builtin('record_move', record_move)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual (len(solutions), 1)
self.assertEqual (len(recorded_moves), 7)
#@unittest.skip("temporarily disabled")
def test_custom_builtin_multiple_bindings(self):
self.rt.register_builtin('multi_binder', multi_binder)
clause = self.parser.parse_line_clause_body('multi_binder(X,Y)')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual (len(solutions), 4)
def _custom_directive(self, db, module_name, clause, user_data):
# logging.debug('custom_directive has been run')
self.assertEqual (len(clause.head.args), 3)
self.assertEqual (unicode(clause.head.args[0]), u'abc')
self.assertEqual (clause.head.args[1].f, 42)
self.assertEqual (clause.head.args[2].s, u'foo')
self.directive_mark = True
#@unittest.skip("temporarily disabled")
def test_custom_directives(self):
self.parser.register_directive('custom_directive', self._custom_directive, None)
self.directive_mark = False
# self.parser.compile_file('samples/dir.pl', UNITTEST_MODULE)
clauses = self.parser.parse_line_clauses('custom_directive(abc, 42, \'foo\').')
self.assertEqual (self.directive_mark, True)
class TestZamiaProlog(unittest.TestCase):
def setUp(self):
#
# db, store
#
db_url = 'sqlite:///foo.db'
# setup compiler + environment
self.db = LogicDB(db_url)
self.parser = PrologParser(self.db)
self.rt = PrologRuntime(self.db)
self.db.clear_module(UNITTEST_MODULE)
def tearDown(self):
self.db.close()
# @unittest.skip("temporarily disabled")
def test_parser(self):
error_catched = False
try:
clause = self.parser.parse_line_clause_body(
'say_eoa(en, "Kids are the best')
logging.debug('clause: %s' % clause)
except PrologError as e:
error_catched = True
self.assertEqual(error_catched, True)
# @unittest.skip("temporarily disabled")
def test_parse_line_clauses(self):
line = 'time_span(TE) :- date_time_stamp(+(D, 1.0)).'
tree = self.parser.parse_line_clauses(line)
logging.debug(unicode(tree[0].body))
self.assertEqual(tree[0].body.name, 'date_time_stamp')
self.assertEqual(tree[0].head.name, 'time_span')
line = 'time_span(tomorrow, TS, TE) :- context(currentTime, T), stamp_date_time(T, date(Y, M, D, H, Mn, S, "local")), date_time_stamp(date(Y, M, +(D, 1.0), 0.0, 0.0, 0.0, "local"), TS), date_time_stamp(date(Y, M, +(D, 1.0), 23.0, 59.0, 59.0, "local"), TE).'
tree = self.parser.parse_line_clauses(line)
logging.debug(unicode(tree[0].body))
self.assertEqual(tree[0].head.name, 'time_span')
self.assertEqual(tree[0].body.name, 'and')
self.assertEqual(len(tree[0].body.args), 4)
# @unittest.skip("temporarily disabled")
def test_kb1(self):
self.assertEqual(len(self.db.lookup('party', 0)), 0)
self.parser.compile_file('samples/kb1.pl', UNITTEST_MODULE)
self.assertEqual(len(self.db.lookup('party', 0)), 1)
clause = self.parser.parse_line_clause_body('woman(X)')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 3)
clause = self.parser.parse_line_clause_body('party')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
clause = self.parser.parse_line_clause_body('woman(fred)')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 0)
# @unittest.skip("temporarily disabled")
def test_parse_to_string(self):
line = u'time_span(c, X, Y) :- p1(c), p2(X, Y); p3(c); p4.'
line2 = u'time_span(c, X, Y) :- or(and(p1(c), p2(X, Y)), p3(c), p4).'
tree = self.parser.parse_line_clauses(line)
logging.debug(unicode(tree[0].body))
self.assertEqual(unicode(tree[0]), line2)
# @unittest.skip("temporarily disabled")
def test_or(self):
self.parser.compile_file('samples/or_test.pl', UNITTEST_MODULE)
# self.rt.set_trace(True)
solutions = self.rt.search_predicate('woman', ['X'])
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 3)
solutions = self.rt.search_predicate('human', ['X'])
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 8)
def test_or_toplevel(self):
self.parser.compile_file('samples/or_test.pl', UNITTEST_MODULE)
clause = self.parser.parse_line_clause_body(
u'woman(mary); woman(jody)')
logging.debug(u'clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
def test_or_bindings(self):
clause = self.parser.parse_line_clause_body(
u'S is "a", or(str_append(S, "b"), str_append(S, "c"))')
logging.debug(u'clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 2)
self.assertEqual(solutions[0]['S'].s, "ab")
self.assertEqual(solutions[1]['S'].s, "ac")
clause = self.parser.parse_line_clause_body(u'X is 42; X is 23')
logging.debug(u'clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 2)
def test_var_access(self):
# set var X from python:
clause = self.parser.parse_line_clause_body('Y is X*X')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause, {'X': NumberLiteral(3)})
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
# access prolog result Y from python:
self.assertEqual(solutions[0]['Y'].f, 9)
def test_list_equality(self):
clause = self.parser.parse_line_clause_body('[] is []')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause, {})
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
clause = self.parser.parse_line_clause_body('[1] is []')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause, {})
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 0)
clause = self.parser.parse_line_clause_body('909442800.0 is []')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause, {})
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 0)
clause = self.parser.parse_line_clause_body('[1,2,3] = [1,2,3]')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause, {})
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
clause = self.parser.parse_line_clause_body('[1,2,3] \\= [1,2,3,4,5]')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause, {})
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
def test_is(self):
clause = self.parser.parse_line_clause_body(
'GENDER is "blubber", GENDER is wde:Male')
logging.debug('clause: %s' % clause)
solutions = self.rt.search(clause, {})
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 0)
# @unittest.skip("temporarily disabled")
def test_list_eval(self):
clause = self.parser.parse_line_clause_body(
'X is 23, Z is 42, Y is [X, U, Z].')
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
self.assertEqual(len(solutions[0]['Y'].l), 3)
self.assertEqual(solutions[0]['Y'].l[0].f, 23.0)
self.assertTrue(isinstance(solutions[0]['Y'].l[1], Variable))
self.assertEqual(solutions[0]['Y'].l[2].f, 42.0)
def test_clauses_location(self):
# this will trigger a runtime error since a(Y) is a predicate,
# but format_str requires a literal arg
clause = self.parser.parse_line_clause_body(
'X is format_str("%s", a(Y))')
logging.debug('clause: %s' % clause)
try:
solutions = self.rt.search(clause, {})
self.fail("we should have seen a runtime error here")
except PrologRuntimeError as e:
self.assertEqual(e.location.line, 1)
self.assertEqual(e.location.col, 29)
def test_cut(self):
self.parser.compile_file('samples/cut_test.pl', UNITTEST_MODULE)
# self.rt.set_trace(True)
clause = self.parser.parse_line_clause_body(u'bar(R, X)')
logging.debug(u'clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 4)
self.assertEqual(solutions[0]['R'].s, "one")
self.assertEqual(solutions[1]['R'].s, "two")
self.assertEqual(solutions[2]['R'].s, "many")
self.assertEqual(solutions[3]['R'].s, "many")
# @unittest.skip("temporarily disabled")
def test_anon_var(self):
clause = self.parser.parse_line_clause_body('_ is 23, _ is 42.')
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
def test_nonvar(self):
clause = self.parser.parse_line_clause_body(u'S is "a", nonvar(S)')
logging.debug(u'clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
clause = self.parser.parse_line_clause_body(u'nonvar(S)')
logging.debug(u'clause: %s' % clause)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 0)
def test_unify_pseudo(self):
clause = self.parser.parse_line_clause_body(
u'C is foo, assertz(mem(foo, bar)), if var(C:mem|bar) then C:mem|bar := 23 endif, X := C:mem|bar'
)
logging.debug(u'clause: %s' % clause)
# self.rt.set_trace(True)
solutions = self.rt.search(clause)
logging.debug('solutions: %s' % repr(solutions))
self.assertEqual(len(solutions), 1)
self.assertEqual(solutions[0]['X'].f, 23.0)