def test_est_query(self):
init_tree()
att_trees = [ATT_TREE, ATT_TREE]
result = [['a1', ['A', 'b1', 'b2']],
['a1', ['A', 'b1']],
['a2', ['A', 'b1', 'b2']],
['a2', ['A', 'b2']]]
gen_data = get_result_cover(att_trees, result)
est = est_query(gen_data, [0, 1], [['a1', 'a2'], [['a2', 'b1'], ['a1', 'b1', 'b2']]])
self.assertEqual(est, 2.5)
def test_get_result_cover(self):
init_tree()
att_trees = [ATT_TREE, ATT_TREE]
result = [['a1', ['A', 'b1', 'b2']],
['a1', ['A', 'b1']],
['a2', ['A', 'b1', 'b2']],
['a2', ['A', 'b2']]]
gen_data = get_result_cover(att_trees, result)
temp = [[{'a1': 1.0}, {'a1': 0.5, 'a2': 0.5, 'b1': 1.0, 'b2': 1.0}],
[{'a1': 1.0}, {'a1': 0.5, 'a2': 0.5, 'b1': 1.0}],
[{'a2': 1.0}, {'a1': 0.5, 'a2': 0.5, 'b1': 1.0, 'b2': 1.0}],
[{'a2': 1.0}, {'a1': 0.5, 'a2': 0.5, 'b2': 1.0}]]
self.assertEqual(gen_data, temp)
