import os, sys
curr_dir = os.path.dirname(os.path.realpath(__file__))
gdl_path = os.path.abspath(os.path.join(curr_dir, os.pardir))
sys.path.append(gdl_path)
from gdl import Lexer, Parser, Database, DatalogError, ParseError
DEFINE = 'GDL> '
QUERY = 'GDL? '
prompt = DEFINE
database = Database()
for filename in sys.argv[1:]:
# read in any files from the command line
with open(filename, 'r') as file:
tokens = Lexer.run_lex(file=file)
for tree in Parser.run_parse(tokens):
# print(tree)
database.define(tree)
try:
# interactive loop
while 1:
raw_string = input(prompt).strip()
if raw_string == '?':
prompt = QUERY
continue
elif raw_string == '.':
prompt = DEFINE
continue
elif not raw_string:
def setUp(self):
self.db = Database()
# FACTS
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('a', 'b', 'c')])
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('x', 'y', 'z')])
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('x', 'y', 'x')])
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('a', 'a', 'a')])
self.db.define_fact('bar', 2, [make_mock_node('1'),
make_mock_node('x', [make_mock_node(x) for x in ('2', '3')])])
# RULES
# Datalog 2.5
# > path(X,Y) :- path(X,Z), link(Z,Y).
# > path(X,Y) :- link(X,Y).
# > link(3,4).
# > link(2,3).
# > link(1,2).
path = make_mock_node('path', [make_mock_node(x) for x in ('?x', '?z')])
link = make_mock_node('link', [make_mock_node(x) for x in ('?z', '?y')])
self.db.define_rule('path', 2, [make_mock_node(x) for x in ('?x', '?y')], [path, link])
link = make_mock_node('link', [make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('path', 2, [make_mock_node(x) for x in ('?x', '?y')], [link])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('3', '4')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('2', '3')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('1', '2')])
# NEGATION
# > x(1). x(2). x(3). x(4).
# > not_path(X, Y) :- x(X), x(Y), not(path(X, Y)).
self.db.define_fact('x', 1, [make_mock_node('1')])
self.db.define_fact('x', 1, [make_mock_node('2')])
self.db.define_fact('x', 1, [make_mock_node('3')])
self.db.define_fact('x', 1, [make_mock_node('4')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
path = make_mock_node('path', [make_mock_node(x) for x in ('?y', '?x')])
self.db.define_rule('rpath', 2, [make_mock_node(x) for x in ('?x', '?y')], [xx, xy, path])
not_path = make_mock_node('not', [make_mock_node('path', [make_mock_node(x) for x in ('?x', '?y')])])
self.db.define_rule('not-path', 2, [make_mock_node(x) for x in ('?x', '?y')], [xx, xy, not_path])
# CYCLICAL RECURSION
# Datalog 2.5
# > s(1).
# > s(2).
# > t(1).
# > p(X) :- q(X), s(X).
# > q(X) :- p(X), t(X).
# > q(X) :- t(X).
self.db.define_fact('s', 1, [make_mock_node('1')])
self.db.define_fact('s', 1, [make_mock_node('2')])
self.db.define_fact('t', 1, [make_mock_node('1')])
p, q, s, t = [make_mock_node(x, [make_mock_node('?x')]) for x in ('p', 'q', 's', 't')]
self.db.define_rule('p', 1, [make_mock_node('?x')], [q, s])
self.db.define_rule('q', 1, [make_mock_node('?x')], [p, t])
self.db.define_rule('q', 1, [make_mock_node('?x')], [t])
# DISTINCT
# (<= (diff ?x ?y) (x ?x) (x ?y) (distinct ?x ?y))
distinct = make_mock_node('distinct', [make_mock_node('?x'), make_mock_node('?y')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
self.db.define_rule('diff', 2, [make_mock_node(x) for x in ('?x', '?y')], [xx, xy, distinct])
# OR
# (<= (valid? ?x ?y) (not-path ?x ?y) (or (distinct ?y 4) (distinct ?x 4)))
d1 = make_mock_node('distinct', [make_mock_node('?y'), make_mock_node('4')])
d2 = make_mock_node('distinct', [make_mock_node('?x'), make_mock_node('4')])
or_ = make_mock_node('or', [d1, d2])
not_path = make_mock_node('not-path', [make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('valid?', 2, [make_mock_node(x) for x in ('?x', '?y')], [not_path, or_])
# 0-ARITY RULES
cell = make_mock_node('cell', [make_mock_node(x) for x in ['?m', '?n', 'b']])
self.db.define_rule('open', 0, [], [make_mock_node('true', [cell])])
self.db.define_rule('terminal', 0, [], [make_mock_node('not', [make_mock_node('open')])])
import os, sys
curr_dir = os.path.dirname(os.path.realpath(__file__))
gdl_path = os.path.abspath(os.path.join(curr_dir, os.pardir))
sys.path.append(gdl_path)
from gdl import Lexer, Parser, Database, DatalogError, ParseError
DEFINE = 'GDL> '
QUERY = 'GDL? '
prompt = DEFINE
database = Database()
for filename in sys.argv[1:]:
# read in any files from the command line
with open(filename, 'r') as file:
tokens = Lexer.run_lex(file=file)
for tree in Parser.run_parse(tokens):
# print(tree)
database.define(tree)
try:
# interactive loop
while 1:
raw_string = input(prompt).strip()
if raw_string == '?':
prompt = QUERY
continue
elif raw_string == '.':
prompt = DEFINE
continue
elif not raw_string:
class TestDatabase(unittest.TestCase):
def setUp(self):
self.db = Database()
# FACTS
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('a', 'b', 'c')])
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('x', 'y', 'z')])
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('x', 'y', 'x')])
self.db.define_fact('foo', 3, [make_mock_node(x) for x in ('a', 'a', 'a')])
self.db.define_fact('bar', 2, [make_mock_node('1'),
make_mock_node('x', [make_mock_node(x) for x in ('2', '3')])])
# RULES
# Datalog 2.5
# > path(X,Y) :- path(X,Z), link(Z,Y).
# > path(X,Y) :- link(X,Y).
# > link(3,4).
# > link(2,3).
# > link(1,2).
path = make_mock_node('path', [make_mock_node(x) for x in ('?x', '?z')])
link = make_mock_node('link', [make_mock_node(x) for x in ('?z', '?y')])
self.db.define_rule('path', 2, [make_mock_node(x) for x in ('?x', '?y')], [path, link])
link = make_mock_node('link', [make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('path', 2, [make_mock_node(x) for x in ('?x', '?y')], [link])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('3', '4')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('2', '3')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('1', '2')])
# NEGATION
# > x(1). x(2). x(3). x(4).
# > not_path(X, Y) :- x(X), x(Y), not(path(X, Y)).
self.db.define_fact('x', 1, [make_mock_node('1')])
self.db.define_fact('x', 1, [make_mock_node('2')])
self.db.define_fact('x', 1, [make_mock_node('3')])
self.db.define_fact('x', 1, [make_mock_node('4')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
path = make_mock_node('path', [make_mock_node(x) for x in ('?y', '?x')])
self.db.define_rule('rpath', 2, [make_mock_node(x) for x in ('?x', '?y')], [xx, xy, path])
not_path = make_mock_node('not', [make_mock_node('path', [make_mock_node(x) for x in ('?x', '?y')])])
self.db.define_rule('not-path', 2, [make_mock_node(x) for x in ('?x', '?y')], [xx, xy, not_path])
# CYCLICAL RECURSION
# Datalog 2.5
# > s(1).
# > s(2).
# > t(1).
# > p(X) :- q(X), s(X).
# > q(X) :- p(X), t(X).
# > q(X) :- t(X).
self.db.define_fact('s', 1, [make_mock_node('1')])
self.db.define_fact('s', 1, [make_mock_node('2')])
self.db.define_fact('t', 1, [make_mock_node('1')])
p, q, s, t = [make_mock_node(x, [make_mock_node('?x')]) for x in ('p', 'q', 's', 't')]
self.db.define_rule('p', 1, [make_mock_node('?x')], [q, s])
self.db.define_rule('q', 1, [make_mock_node('?x')], [p, t])
self.db.define_rule('q', 1, [make_mock_node('?x')], [t])
# DISTINCT
# (<= (diff ?x ?y) (x ?x) (x ?y) (distinct ?x ?y))
distinct = make_mock_node('distinct', [make_mock_node('?x'), make_mock_node('?y')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
self.db.define_rule('diff', 2, [make_mock_node(x) for x in ('?x', '?y')], [xx, xy, distinct])
# OR
# (<= (valid? ?x ?y) (not-path ?x ?y) (or (distinct ?y 4) (distinct ?x 4)))
d1 = make_mock_node('distinct', [make_mock_node('?y'), make_mock_node('4')])
d2 = make_mock_node('distinct', [make_mock_node('?x'), make_mock_node('4')])
or_ = make_mock_node('or', [d1, d2])
not_path = make_mock_node('not-path', [make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('valid?', 2, [make_mock_node(x) for x in ('?x', '?y')], [not_path, or_])
# 0-ARITY RULES
cell = make_mock_node('cell', [make_mock_node(x) for x in ['?m', '?n', 'b']])
self.db.define_rule('open', 0, [], [make_mock_node('true', [cell])])
self.db.define_rule('terminal', 0, [], [make_mock_node('not', [make_mock_node('open')])])
def test_fact_list_error(self):
with self.assertRaises(TypeError):
Database().define_fact('foo', 2, (1, 2))
def test_fact_variable_argument_error(self):
children = [make_mock_node('x'), make_mock_node('?y')]
arg = make_mock_node('predicate', children)
with self.assertRaises(DatalogError):
Database().define_fact('foo', 1, [arg])
def test_fact_query_error(self):
children = [make_mock_node('x'), make_mock_node('?x')]
node = make_mock_node('foo', children)
with self.assertRaises(DatalogError):
self.db.query(node)
def test_fact_query_success(self):
args = [make_mock_node(x) for x in ('x', 'y', 'z')]
foo = make_mock_node('foo', args)
self.assertTrue(self.db.query(foo))
def test_fact_query_failure(self):
args = [make_mock_node(x) for x in ('c', 'b', 'a')]
foo = make_mock_node('foo', args)
self.assertFalse(self.db.query(foo))
def test_fact_query_match(self):
answer = [{'?b': 'b', '?c': 'c'}, {'?b': 'a', '?c': 'a'}]
args = [make_mock_node(x) for x in ('a', '?b', '?c')]
foo = make_mock_node('foo', args)
results = [{k: d[k].term for k in d} for d in self.db.query(foo)]
self.assertEqual(answer, results)
def test_fact_query_match_complex(self):
answer = make_mock_node('x', [make_mock_node(x) for x in ('2', '3')])
args = [make_mock_node('1'), make_mock_node('?x')]
bar = make_mock_node('bar', args)
results = self.db.query(bar)
self.assertEqual(results, [{'?x': answer}])
def test_fact_query_repeat_variables(self):
answer = [{'?1': 'x', '?2': 'y'}, {'?1': 'a', '?2': 'a'}]
args = [make_mock_node(x) for x in ('?1', '?2', '?1')]
foo = make_mock_node('foo', args)
results = [{k: d[k].term for k in d} for d in self.db.query(foo)]
self.assertEqual(answer, results)
# > path(1,4)?
# path(1, 4).
def test_rule_success(self):
query = make_mock_node('path', [make_mock_node('1'), make_mock_node('4')])
self.assertTrue(self.db.query(query))
def test_rule_success2(self):
query = make_mock_node('rpath', [make_mock_node('4'), make_mock_node('1')])
self.assertTrue(self.db.query(query))
def test_rule_failure(self):
query = make_mock_node('path', [make_mock_node('4'), make_mock_node('?x')])
self.assertFalse(self.db.query(query))
def test_literal_negation_success(self):
query = make_mock_node('not-path', [make_mock_node('4'), make_mock_node('1')])
self.assertTrue(self.db.query(query))
def test_literal_negation_failure(self):
query = make_mock_node('not-path', [make_mock_node('1'), make_mock_node('3')])
self.assertFalse(self.db.query(query))
# > p(X)?
# p(1).
def test_rule_cyclical_recursion(self):
answer = [{'?x': '1'}]
query = make_mock_node('p', [make_mock_node('?x')])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertEqual(results, answer)
def test_rule_redefinition_deletes_old_facts(self):
answer = [{'?x': '1'}, {'?x': '2'}]
query = make_mock_node('p', [make_mock_node('?x')])
self.db.query(query)
self.db.define_rule('p', 1,
[make_mock_node('?x')],
[make_mock_node('s', [make_mock_node('?x')])])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertEqual(results, answer)
def test_distinct(self):
query = make_mock_node('diff', [make_mock_node('?x'), make_mock_node('?y')])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertEqual(12, len(results))
for i in range(1, 5):
self.assertNotIn({'?x': i, '?y': i}, results)
def test_or(self):
query = make_mock_node('valid?', [make_mock_node('?x'), make_mock_node('?y')])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertNotIn({'?x': 4, '?y': 4}, results)
def test_negative_variable_error(self):
distinct = make_mock_node('distinct', [make_mock_node('?x'), make_mock_node('?y')])
path = make_mock_node('path', [make_mock_node('?z'), make_mock_node('?y')])
body = [path, distinct]
with self.assertRaises(DatalogError):
self.db.define_rule('diff', 2, [make_mock_node('?x'), make_mock_node('?y')], body)
def test_negative_cycle_error(self):
# > p_(X) :- q_(X)
# > r_(X) :- p_(X)
# > q_(X) :- x_(X), ~r_(X)
p = make_mock_node('p_', [make_mock_node('?x')])
q = make_mock_node('q_', [make_mock_node('?x')])
r = make_mock_node('r_', [make_mock_node('?x')])
x = make_mock_node('x_', [make_mock_node('?x')])
self.db.define_rule('p_', 1, [make_mock_node('?x')], [q])
self.db.define_rule('r_', 1, [make_mock_node('?x')], [p])
with self.assertRaises(DatalogError):
self.db.define_rule('q_', 1, [make_mock_node('?x')], [x, make_mock_node('not', [r])])
def test_reserved_word_in_fact_error(self):
not4 = make_mock_node('not', [make_mock_node('4')])
with self.assertRaises(DatalogError):
self.db.define_fact('x', 1, [not4])
def test_reserved_word_in_fact_error(self):
q = make_mock_node('q_', [make_mock_node('?x')])
with self.assertRaises(DatalogError):
self.db.define_rule('p_', 1, [make_mock_node('not', [make_mock_node('?x')])], [q])
def test_define_fact(self):
fact = make_mock_node('new_fact', [make_mock_node(x) for x in ('1', '2', '3')])
self.db.define(fact)
results = self.db.query(make_mock_node('new_fact', [make_mock_node(x) for x in ('?x', '?y', '?z')]))
results = [{k: d[k].term for k in d} for d in results]
self.assertEqual(results, [{'?x': '1', '?y': '2', '?z': '3'}])
def test_define_rule(self):
distinct = make_mock_node('distinct', [make_mock_node('?x'), make_mock_node('1')])
x = make_mock_node('x', [make_mock_node('?x')])
head = make_mock_node('new_rule', [make_mock_node('?x')])
rule = make_mock_node('<=', [head, x, distinct])
self.db.define(rule)
results = self.db.query(make_mock_node('new_rule', [make_mock_node('?x')]))
results = [{k: d[k].term for k in d} for d in results]
self.assertEqual(results, [{'?x': '2'}, {'?x': '3'}, {'?x': '4'}])
def test_0_arity_rules(self):
self.assertFalse(self.db.query(make_mock_node('open')))
self.assertTrue(self.db.query(make_mock_node('terminal')))
cell = make_mock_node('cell', [make_mock_node(x) for x in ('2', '2', 'b')])
self.db.define_fact('true', 1, [cell])
self.assertTrue(self.db.query(make_mock_node('open')))
self.assertFalse(self.db.query(make_mock_node('terminal')))
def test_evaluate_negative_literals_last(self):
xx = make_mock_node('x', [make_mock_node('?x')])
not_s = make_mock_node('not', [make_mock_node('s', [make_mock_node('?x')])])
head = make_mock_node('not-y', [make_mock_node('?x')])
rule = make_mock_node('<=', [head, not_s, xx])
self.db.define(rule)
results = self.db.query(make_mock_node('not-y', [make_mock_node('?x')]))
results = [{k: d[k].term for k in d} for d in results]
self.assertEqual(results, [{'?x': '3'}, {'?x': '4'}])
def setUp(self):
self.db = Database()
# FACTS
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('a', 'b', 'c')])
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('x', 'y', 'z')])
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('x', 'y', 'x')])
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('a', 'a', 'a')])
self.db.define_fact('bar', 2, [
make_mock_node('1'),
make_mock_node('x', [make_mock_node(x) for x in ('2', '3')])
])
# RULES
# Datalog 2.5
# > path(X,Y) :- path(X,Z), link(Z,Y).
# > path(X,Y) :- link(X,Y).
# > link(3,4).
# > link(2,3).
# > link(1,2).
path = make_mock_node('path',
[make_mock_node(x) for x in ('?x', '?z')])
link = make_mock_node('link',
[make_mock_node(x) for x in ('?z', '?y')])
self.db.define_rule('path', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[path, link])
link = make_mock_node('link',
[make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('path', 2,
[make_mock_node(x) for x in ('?x', '?y')], [link])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('3', '4')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('2', '3')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('1', '2')])
# NEGATION
# > x(1). x(2). x(3). x(4).
# > not_path(X, Y) :- x(X), x(Y), not(path(X, Y)).
self.db.define_fact('x', 1, [make_mock_node('1')])
self.db.define_fact('x', 1, [make_mock_node('2')])
self.db.define_fact('x', 1, [make_mock_node('3')])
self.db.define_fact('x', 1, [make_mock_node('4')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
path = make_mock_node('path',
[make_mock_node(x) for x in ('?y', '?x')])
self.db.define_rule('rpath', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[xx, xy, path])
not_path = make_mock_node('not', [
make_mock_node('path', [make_mock_node(x) for x in ('?x', '?y')])
])
self.db.define_rule('not-path', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[xx, xy, not_path])
# CYCLICAL RECURSION
# Datalog 2.5
# > s(1).
# > s(2).
# > t(1).
# > p(X) :- q(X), s(X).
# > q(X) :- p(X), t(X).
# > q(X) :- t(X).
self.db.define_fact('s', 1, [make_mock_node('1')])
self.db.define_fact('s', 1, [make_mock_node('2')])
self.db.define_fact('t', 1, [make_mock_node('1')])
p, q, s, t = [
make_mock_node(x, [make_mock_node('?x')])
for x in ('p', 'q', 's', 't')
]
self.db.define_rule('p', 1, [make_mock_node('?x')], [q, s])
self.db.define_rule('q', 1, [make_mock_node('?x')], [p, t])
self.db.define_rule('q', 1, [make_mock_node('?x')], [t])
# DISTINCT
# (<= (diff ?x ?y) (x ?x) (x ?y) (distinct ?x ?y))
distinct = make_mock_node(
'distinct',
[make_mock_node('?x'), make_mock_node('?y')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
self.db.define_rule('diff', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[xx, xy, distinct])
# OR
# (<= (valid? ?x ?y) (not-path ?x ?y) (or (distinct ?y 4) (distinct ?x 4)))
d1 = make_mock_node(
'distinct',
[make_mock_node('?y'), make_mock_node('4')])
d2 = make_mock_node(
'distinct',
[make_mock_node('?x'), make_mock_node('4')])
or_ = make_mock_node('or', [d1, d2])
not_path = make_mock_node('not-path',
[make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('valid?', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[not_path, or_])
# 0-ARITY RULES
cell = make_mock_node('cell',
[make_mock_node(x) for x in ['?m', '?n', 'b']])
self.db.define_rule('open', 0, [], [make_mock_node('true', [cell])])
self.db.define_rule('terminal', 0, [],
[make_mock_node('not', [make_mock_node('open')])])
class TestDatabase(unittest.TestCase):
def setUp(self):
self.db = Database()
# FACTS
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('a', 'b', 'c')])
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('x', 'y', 'z')])
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('x', 'y', 'x')])
self.db.define_fact('foo', 3,
[make_mock_node(x) for x in ('a', 'a', 'a')])
self.db.define_fact('bar', 2, [
make_mock_node('1'),
make_mock_node('x', [make_mock_node(x) for x in ('2', '3')])
])
# RULES
# Datalog 2.5
# > path(X,Y) :- path(X,Z), link(Z,Y).
# > path(X,Y) :- link(X,Y).
# > link(3,4).
# > link(2,3).
# > link(1,2).
path = make_mock_node('path',
[make_mock_node(x) for x in ('?x', '?z')])
link = make_mock_node('link',
[make_mock_node(x) for x in ('?z', '?y')])
self.db.define_rule('path', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[path, link])
link = make_mock_node('link',
[make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('path', 2,
[make_mock_node(x) for x in ('?x', '?y')], [link])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('3', '4')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('2', '3')])
self.db.define_fact('link', 2, [make_mock_node(x) for x in ('1', '2')])
# NEGATION
# > x(1). x(2). x(3). x(4).
# > not_path(X, Y) :- x(X), x(Y), not(path(X, Y)).
self.db.define_fact('x', 1, [make_mock_node('1')])
self.db.define_fact('x', 1, [make_mock_node('2')])
self.db.define_fact('x', 1, [make_mock_node('3')])
self.db.define_fact('x', 1, [make_mock_node('4')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
path = make_mock_node('path',
[make_mock_node(x) for x in ('?y', '?x')])
self.db.define_rule('rpath', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[xx, xy, path])
not_path = make_mock_node('not', [
make_mock_node('path', [make_mock_node(x) for x in ('?x', '?y')])
])
self.db.define_rule('not-path', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[xx, xy, not_path])
# CYCLICAL RECURSION
# Datalog 2.5
# > s(1).
# > s(2).
# > t(1).
# > p(X) :- q(X), s(X).
# > q(X) :- p(X), t(X).
# > q(X) :- t(X).
self.db.define_fact('s', 1, [make_mock_node('1')])
self.db.define_fact('s', 1, [make_mock_node('2')])
self.db.define_fact('t', 1, [make_mock_node('1')])
p, q, s, t = [
make_mock_node(x, [make_mock_node('?x')])
for x in ('p', 'q', 's', 't')
]
self.db.define_rule('p', 1, [make_mock_node('?x')], [q, s])
self.db.define_rule('q', 1, [make_mock_node('?x')], [p, t])
self.db.define_rule('q', 1, [make_mock_node('?x')], [t])
# DISTINCT
# (<= (diff ?x ?y) (x ?x) (x ?y) (distinct ?x ?y))
distinct = make_mock_node(
'distinct',
[make_mock_node('?x'), make_mock_node('?y')])
xx = make_mock_node('x', [make_mock_node('?x')])
xy = make_mock_node('x', [make_mock_node('?y')])
self.db.define_rule('diff', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[xx, xy, distinct])
# OR
# (<= (valid? ?x ?y) (not-path ?x ?y) (or (distinct ?y 4) (distinct ?x 4)))
d1 = make_mock_node(
'distinct',
[make_mock_node('?y'), make_mock_node('4')])
d2 = make_mock_node(
'distinct',
[make_mock_node('?x'), make_mock_node('4')])
or_ = make_mock_node('or', [d1, d2])
not_path = make_mock_node('not-path',
[make_mock_node(x) for x in ('?x', '?y')])
self.db.define_rule('valid?', 2,
[make_mock_node(x) for x in ('?x', '?y')],
[not_path, or_])
# 0-ARITY RULES
cell = make_mock_node('cell',
[make_mock_node(x) for x in ['?m', '?n', 'b']])
self.db.define_rule('open', 0, [], [make_mock_node('true', [cell])])
self.db.define_rule('terminal', 0, [],
[make_mock_node('not', [make_mock_node('open')])])
def test_fact_list_error(self):
with self.assertRaises(TypeError):
Database().define_fact('foo', 2, (1, 2))
def test_fact_variable_argument_error(self):
children = [make_mock_node('x'), make_mock_node('?y')]
arg = make_mock_node('predicate', children)
with self.assertRaises(DatalogError):
Database().define_fact('foo', 1, [arg])
def test_fact_query_error(self):
children = [make_mock_node('x'), make_mock_node('?x')]
node = make_mock_node('foo', children)
with self.assertRaises(DatalogError):
self.db.query(node)
def test_fact_query_success(self):
args = [make_mock_node(x) for x in ('x', 'y', 'z')]
foo = make_mock_node('foo', args)
self.assertTrue(self.db.query(foo))
def test_fact_query_failure(self):
args = [make_mock_node(x) for x in ('c', 'b', 'a')]
foo = make_mock_node('foo', args)
self.assertFalse(self.db.query(foo))
def test_fact_query_match(self):
answer = [{'?b': 'b', '?c': 'c'}, {'?b': 'a', '?c': 'a'}]
args = [make_mock_node(x) for x in ('a', '?b', '?c')]
foo = make_mock_node('foo', args)
results = [{k: d[k].term for k in d} for d in self.db.query(foo)]
self.assertEqual(answer, results)
def test_fact_query_match_complex(self):
answer = make_mock_node('x', [make_mock_node(x) for x in ('2', '3')])
args = [make_mock_node('1'), make_mock_node('?x')]
bar = make_mock_node('bar', args)
results = self.db.query(bar)
self.assertEqual(results, [{'?x': answer}])
def test_fact_query_repeat_variables(self):
answer = [{'?1': 'x', '?2': 'y'}, {'?1': 'a', '?2': 'a'}]
args = [make_mock_node(x) for x in ('?1', '?2', '?1')]
foo = make_mock_node('foo', args)
results = [{k: d[k].term for k in d} for d in self.db.query(foo)]
self.assertEqual(answer, results)
# > path(1,4)?
# path(1, 4).
def test_rule_success(self):
query = make_mock_node(
'path',
[make_mock_node('1'), make_mock_node('4')])
self.assertTrue(self.db.query(query))
def test_rule_success2(self):
query = make_mock_node(
'rpath',
[make_mock_node('4'), make_mock_node('1')])
self.assertTrue(self.db.query(query))
def test_rule_failure(self):
query = make_mock_node(
'path',
[make_mock_node('4'), make_mock_node('?x')])
self.assertFalse(self.db.query(query))
def test_literal_negation_success(self):
query = make_mock_node(
'not-path',
[make_mock_node('4'), make_mock_node('1')])
self.assertTrue(self.db.query(query))
def test_literal_negation_failure(self):
query = make_mock_node(
'not-path',
[make_mock_node('1'), make_mock_node('3')])
self.assertFalse(self.db.query(query))
# > p(X)?
# p(1).
def test_rule_cyclical_recursion(self):
answer = [{'?x': '1'}]
query = make_mock_node('p', [make_mock_node('?x')])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertEqual(results, answer)
def test_rule_redefinition_deletes_old_facts(self):
answer = [{'?x': '1'}, {'?x': '2'}]
query = make_mock_node('p', [make_mock_node('?x')])
self.db.query(query)
self.db.define_rule('p', 1, [make_mock_node('?x')],
[make_mock_node('s', [make_mock_node('?x')])])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertEqual(results, answer)
def test_distinct(self):
query = make_mock_node(
'diff',
[make_mock_node('?x'), make_mock_node('?y')])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertEqual(12, len(results))
for i in range(1, 5):
self.assertNotIn({'?x': i, '?y': i}, results)
def test_or(self):
query = make_mock_node(
'valid?',
[make_mock_node('?x'), make_mock_node('?y')])
results = [{k: d[k].term for k in d} for d in self.db.query(query)]
self.assertNotIn({'?x': 4, '?y': 4}, results)
def test_negative_variable_error(self):
distinct = make_mock_node(
'distinct',
[make_mock_node('?x'), make_mock_node('?y')])
path = make_mock_node(
'path',
[make_mock_node('?z'), make_mock_node('?y')])
body = [path, distinct]
with self.assertRaises(DatalogError):
self.db.define_rule('diff', 2,
[make_mock_node('?x'),
make_mock_node('?y')], body)
def test_negative_cycle_error(self):
# > p_(X) :- q_(X)
# > r_(X) :- p_(X)
# > q_(X) :- x_(X), ~r_(X)
p = make_mock_node('p_', [make_mock_node('?x')])
q = make_mock_node('q_', [make_mock_node('?x')])
r = make_mock_node('r_', [make_mock_node('?x')])
x = make_mock_node('x_', [make_mock_node('?x')])
self.db.define_rule('p_', 1, [make_mock_node('?x')], [q])
self.db.define_rule('r_', 1, [make_mock_node('?x')], [p])
with self.assertRaises(DatalogError):
self.db.define_rule('q_', 1, [make_mock_node('?x')],
[x, make_mock_node('not', [r])])
def test_reserved_word_in_fact_error(self):
not4 = make_mock_node('not', [make_mock_node('4')])
with self.assertRaises(DatalogError):
self.db.define_fact('x', 1, [not4])
def test_reserved_word_in_fact_error(self):
q = make_mock_node('q_', [make_mock_node('?x')])
with self.assertRaises(DatalogError):
self.db.define_rule(
'p_', 1, [make_mock_node('not', [make_mock_node('?x')])], [q])
def test_define_fact(self):
fact = make_mock_node('new_fact',
[make_mock_node(x) for x in ('1', '2', '3')])
self.db.define(fact)
results = self.db.query(
make_mock_node('new_fact',
[make_mock_node(x) for x in ('?x', '?y', '?z')]))
results = [{k: d[k].term for k in d} for d in results]
self.assertEqual(results, [{'?x': '1', '?y': '2', '?z': '3'}])
def test_define_rule(self):
distinct = make_mock_node(
'distinct',
[make_mock_node('?x'), make_mock_node('1')])
x = make_mock_node('x', [make_mock_node('?x')])
head = make_mock_node('new_rule', [make_mock_node('?x')])
rule = make_mock_node('<=', [head, x, distinct])
self.db.define(rule)
results = self.db.query(
make_mock_node('new_rule', [make_mock_node('?x')]))
results = [{k: d[k].term for k in d} for d in results]
self.assertEqual(results, [{'?x': '2'}, {'?x': '3'}, {'?x': '4'}])
def test_0_arity_rules(self):
self.assertFalse(self.db.query(make_mock_node('open')))
self.assertTrue(self.db.query(make_mock_node('terminal')))
cell = make_mock_node('cell',
[make_mock_node(x) for x in ('2', '2', 'b')])
self.db.define_fact('true', 1, [cell])
self.assertTrue(self.db.query(make_mock_node('open')))
self.assertFalse(self.db.query(make_mock_node('terminal')))
def test_evaluate_negative_literals_last(self):
xx = make_mock_node('x', [make_mock_node('?x')])
not_s = make_mock_node('not',
[make_mock_node('s', [make_mock_node('?x')])])
head = make_mock_node('not-y', [make_mock_node('?x')])
rule = make_mock_node('<=', [head, not_s, xx])
self.db.define(rule)
results = self.db.query(make_mock_node('not-y',
[make_mock_node('?x')]))
results = [{k: d[k].term for k in d} for d in results]
self.assertEqual(results, [{'?x': '3'}, {'?x': '4'}])
def test_fact_variable_argument_error(self):
children = [make_mock_node('x'), make_mock_node('?y')]
arg = make_mock_node('predicate', children)
with self.assertRaises(DatalogError):
Database().define_fact('foo', 1, [arg])
def test_fact_list_error(self):
with self.assertRaises(TypeError):
Database().define_fact('foo', 2, (1, 2))