Example #1
0
    def test_walk_given_recursive_block(self):
        as_tree = ast.parse(ms("""\
             z = 2           # 0th block
             while a < 3:    # 1st block
                 if a < 2:   # 2nd block
                      z = 2  # 3rd block
                 b = 2       # 4th block
             c = 3           # 5th block
             """))

        cfg_real = Cfg(as_tree)
        blocks_list = []
        for blocks in cfg_real.walk_block(cfg_real.block_list[0]):
            blocks_list.append(blocks)
        pass
Example #2
0
    def test_walk_given_2_block(self):
        as_tree = ast.parse(ms("""\
            z = 2
            if z < 2:
                y = 3
            """))

        cfg_real = Cfg(as_tree)
        # cfg_real.walk_block(cfg_real.block_list[0])

        blocks_list = []
        for blocks in cfg_real.walk_block(cfg_real.block_list[0]):
            blocks_list.append(blocks)

        expected_block_list = build_blocks([3, 3, None], [1, 2, 'If'],
                                                block_links={'1': [0], '0':[]})

        self.assertBasicBlockListEqual(blocks_list, expected_block_list)