def test_generate_schema(self):
uids = [
UID("hasChild", "hasChild-"),
UID("birthday", "hasChild-")
]
functions = []
function = Function("GetParent")
function.add_atom("hasChild-", "x", "y")
functions.append(function)
function = Function("GetGrandChildrenBirthday")
function.add_atom("hasChild", "x", "y")
function.add_atom("hasChild", "y", "z")
function.add_atom("birthday", "z", "t")
function.set_existential_variable("y")
functions.append(function)
relations = ["hasChild", "hasChild-", "birthday", "birthday-"]
schema = get_schema_xml(functions, relations, uids)
self.assertIn("schema", schema)
class TestFunction(unittest.TestCase):
def setUp(self) -> None:
self.function = Function()
def test_create_empty_function(self):
self.assertEqual(self.function.get_number_variables(), 0)
self.assertEqual(self.function.get_number_atoms(), 0)
def test_add_atom(self):
self.function.add_atom("r", "x", "x")
self.assertEqual(self.function.get_number_variables(), 1)
self.assertEqual(self.function.get_number_atoms(), 1)
self.function.add_atom("r", "y", "y")
self.assertEqual(self.function.get_number_variables(), 2)
self.assertEqual(self.function.get_number_atoms(), 2)
self.function.add_atom("r", "x", "y")
self.assertEqual(self.function.get_number_variables(), 2)
self.function.add_atom("r", "t", "u")
self.assertEqual(self.function.get_number_variables(), 4)
def test_get_linear_paths(self):
self.assertEqual(self.function.get_linear_paths(), [])
self.function.add_atom("r1", "x", "y")
self.assertEqual(sorted(self.function.get_linear_paths()),
sorted([["r1"]]))
self.function.add_atom("r2", "y", "z")
self.assertEqual(sorted(self.function.get_linear_paths()),
sorted([["r1"], ["r1", "r2"]]))
self.function.add_atom("r3", "x", "z")
self.assertEqual(sorted(self.function.get_linear_paths()),
sorted([["r1"], ["r1", "r2"], ["r3"], ["r3", "r2-"]]))
self.function.add_atom("r4-", "z", "y")
self.assertEqual(
sorted(self.function.get_linear_paths()),
sorted([["r1"], ["r1", "r2"], ["r3"], ["r3", "r2-"], ["r1", "r4"],
["r3", "r4-"]]))
def test_real_linear(self):
self.function.add_atom("r1", "x", "y")
self.function.add_atom("r2", "y", "z")
self.assertTrue(self.function.is_linear())
def test_false_linear(self):
self.function.add_atom("r1", "x", "y")
self.function.add_atom("r2", "y", "z")
self.function.add_atom("r2", "x", "z")
self.assertFalse(self.function.is_linear())
def test_false_linear2(self):
self.function.add_atom("r1", "x", "y")
self.function.add_atom("r2", "z", "t")
self.assertFalse(self.function.is_linear())
def test_get_all_linear_subfunctions(self):
self.function.add_atom("r1", "x", "y")
self.function.add_atom("r2", "y", "z")
self.function.add_atom("r3", "y", "t")
self.assertEqual(len(self.function.get_linear_subfunctions()), 3)
for function in self.function.get_linear_subfunctions():
self.assertTrue(function.is_linear())
def test_get_xml_relation(self):
self.function.name = "A"
self.function.add_atom("r1", "xxx", "yyy")
self.function.add_atom("r2", "yyy", "zzz")
self.function.add_atom("r2", "zzz", "ttt")
self.function.set_existential_variable("zzz")
xml = self.function.get_xml_relation()
self.assertIn("View_A", xml)
self.assertIn("xxx", xml)
self.assertIn("yyy", xml)
self.assertNotIn("zzz", xml)
self.assertIn("ttt", xml)
def test_get_xml_dependencies(self):
self.function.name = "A"
self.function.add_atom("r1", "xxx", "yyy")
self.function.add_atom("r2", "yyy", "zzz")
self.function.add_atom("r2", "zzz", "ttt")
self.function.set_existential_variable("zzz")
xml = self.function.get_xml_dependencies()
self.assertEqual(xml.count("View_A"), 2)
self.assertEqual(xml.count("xxx"), 4)
self.assertEqual(xml.count("yyy"), 6)
self.assertEqual(xml.count("ttt"), 4)
self.assertEqual(xml.count("zzz"), 4)
def test_existential(self):
self.assertEqual(self.function.get_number_existential_variables(), 0)
self.function.add_atom("r1", "x", "y")
self.function.set_existential_variable("y")
self.assertEqual(self.function.get_linear_paths(), [])
self.function.add_atom("r2", "y", "z")
self.assertEqual(sorted(self.function.get_linear_paths()),
sorted([["r1", "r2"]]))
def test_set_input_variable(self):
self.function.add_atom("r1", "x", "y")
self.function.set_input_variable("y")
self.assertEqual(self.function.get_linear_paths(), [["r1-"]])
def test_get_longest_query(self):
self.assertEqual(self.function.get_longest_query(), [])
self.function.add_atom("r1", "x", "y")
self.assertEqual(self.function.get_longest_query(), ["r1"])
def test_grammar(self):
self.function.add_atom("r1", "x", "y")
relations = ["r1", "r2", "r1-", "r2-"]
uids = [UID("r1", "r2")]
cfg = self.function.get_longest_query_grammar(relations, uids)
self.assertTrue(cfg.contains(["r1"]))
self.assertTrue(cfg.contains(["r2", "r2-", "r1"]))
self.assertFalse(cfg.contains(["r2", "r1"]))
def test_get_relations_and_inverses(self):
self.assertEqual(self.function.get_relations_and_inverses(), set())
self.function.add_atom("r1", "x", "y")
self.assertEqual(self.function.get_relations_and_inverses(),
{"r1", "r1-"})