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)