def test_get_simple_block_given_no_indent(self):
as_tree = ast.parse(ms("""\
a = 3
a = 4
""")
)
cfg_holder = Cfg()
simple_block_list = []
for simple_block in cfg_holder.get_basic_block(as_tree.body):
simple_block_list.append(simple_block)
expected_block = RawBasicBlock(start_line=1, end_line=2)
self.assertBasicBlockEqual(simple_block_list[0], expected_block)
def test_get_simple_block_given_if(self):
as_tree = ast.parse(ms("""\
a = 3
if a < 3:
b = 4
c = 5
""")
)
cfg_holder = Cfg()
simple_block_list = []
for basic_block in cfg_holder.get_basic_block(as_tree.body):
simple_block_list.append(basic_block)
expected_block_0 = RawBasicBlock(start_line=1, end_line=2, block_end_type='If')
expected_block_1 = RawBasicBlock(start_line=4, end_line=4)
self.assertBasicBlockEqual(simple_block_list[0], expected_block_0, block_index=0)
self.assertBasicBlockEqual(simple_block_list[1], expected_block_1, block_index=1)
