def test_star(self): root = tree.build_tree(STAR) self.assertEquals(root.name, 'Sort') hash_agg = root.children[0] self.assertEquals(len(root.children), 1) self.assertEquals(hash_agg.name, 'HashAggregate') self.assertEquals([x.name for x in hash_agg.children], ['Nested Loop']) parent = hash_agg.children[0] for names in [ ['Hash Join', 'Index Scan'], ['Hash Join', 'Hash'],['Append', 'Hash'], ['Seq Scan', 'Seq Scan', 'Seq Scan', 'Seq Scan', 'Seq Scan', ]]: self.assertEquals([x.name for x in parent.children], names) parent = parent.children[0] nest_loop = hash_agg.children[0]
def test_basic(self): root = tree.build_tree(BASIC) self.assertEquals(root.name, "Result")