def test_compute_liveout_with_attr(self):
blocks, as_tree = self.build_arbitrary_blocks(
block_links={"A": ["B"], "B": []},
code={
"A": dedent(
"""\
a = 4
a.b = 2
"""
),
"B": dedent(
"""\
c = a.b
"""
),
},
)
cfg_real = Cfg()
cfg_real.block_list = blocks
cfg_real.as_tree = as_tree
cfg_real.root = cfg_real.block_list[0]
cfg_real.gather_initial_info()
cfg_real.compute_live_out()
expected_live_out = {"A": {"a", "a.b"}, "B": set()}
self.assertLiveOutEqual(cfg_real.block_list, expected_live_out)
def test_compute_liveout_given_5_blocks(self):
blocks, as_tree = self.build_arbitrary_blocks(
block_links={"A": ["B"], "B": ["C", "D"], "C": ["D"], "D": ["E", "B"], "E": []},
code={
"A": dedent(
"""\
i = 1
"""
),
"B": dedent(
"""\
if i < 0:
pass
"""
),
"C": dedent(
"""\
s = 0
"""
),
"D": dedent(
"""\
s = s + i
i = i + 1
if i < 0:
pass
"""
),
"E": dedent(
"""\
if s < 3:
pass
"""
),
},
)
cfg_real = Cfg()
cfg_real.block_list = blocks
cfg_real.as_tree = as_tree
cfg_real.root = cfg_real.block_list[0]
cfg_real.gather_initial_info()
cfg_real.compute_live_out()
expected_live_out = {"A": {"s", "i"}, "B": {"s", "i"}, "C": {"s", "i"}, "D": {"s", "i"}, "E": set()}
self.assertLiveOutEqual(cfg_real.block_list, expected_live_out)
def test_insert_phi_function_pruned_4_blocks(self):
blocks, as_tree = self.build_arbitrary_blocks(
block_links={"A": ["B", "C"], "B": ["D"], "C": ["D"], "D": []},
code={
"A": dedent(
"""\
pass
"""
),
"B": dedent(
"""\
var = 3
"""
),
"C": dedent(
"""\
pass
"""
),
"D": dedent(
"""\
if var < 3:
pass
"""
),
},
block_type={"A": ParentScopeBlock},
)
cfg_real = Cfg()
cfg_real.block_list = blocks
cfg_real.as_tree = as_tree
cfg_real.root = cfg_real.block_list[0]
cfg_real.root.blocks = blocks
cfg_real.fill_df()
cfg_real.gather_initial_info()
cfg_real.compute_live_out()
cfg_real.ins_phi_function_all()
expected_phi_list = {"A": set(), "B": set(), "C": set(), "D": {"var"}}
self.assertPhiListEqual(cfg_real.block_list, expected_phi_list)
def test_rename_given_custom_4_blocks(self):
r"""
A
/ \
B E
/ \ |
C D |
\ / |
F <----
"""
blocks, as_tree = self.build_arbitrary_blocks(
block_links={
"A": ["B", "E"],
"B": ["C", "D"],
"C": ["F"],
"D": ["F"],
"E": ["G"],
"F": ["G"],
"G": []
},
block_type={"A": ParentScopeBlock},
code={
"A":
dedent("""\
"""),
"B":
dedent("""\
a = 1 #a_0
"""),
"C":
dedent("""\
a = 22 #a_1
"""),
"D":
dedent("""\
a = 33 #a_2
"""),
"E":
dedent("""\
a = 44 #a_4
"""),
"F":
dedent("""\
a = 55 #a_3
"""),
"G":
dedent("""\
a = 66 #a_5
"""),
},
)
cfg_real = Cfg()
cfg_real.block_list = blocks
cfg_real.as_tree = as_tree
cfg_real.root = cfg_real.block_list[0]
cfg_real.root.ast_node = as_tree
cfg_real.root.ssa_code.code_list.append(as_tree)
cfg_real.root.blocks = blocks
cfg_real.fill_df()
cfg_real.gather_initial_info()
cfg_real.compute_live_out()
cfg_real.rename_to_ssa()
self.assertBlockSsaList(
cfg_real.block_list,
{
"A":
dedent("""\
Module
"""),
"B":
dedent("""\
Assign: (a_0,) = 1
"""),
"C":
dedent("""\
Assign: (a_1,) = 22
"""),
"D":
dedent("""\
Assign: (a_2,) = 33
"""),
"E":
dedent("""\
Assign: (a_4,) = 44
"""),
"F":
dedent("""\
Assign: (a_3,) = 55
"""),
"G":
dedent("""\
Assign: (a_5,) = 66
"""),
},
)
def test_insert_phi_function_pruned(self):
r"""
Note: '|' with no arrows means pointing down
A
|
B <------|
/ \ |
C F |
| / \ |
| G I |
| \ / |
| H |
\ / |
D-----------|
|
E
"""
blocks, as_tree = self.build_arbitrary_blocks(
block_links={
"A": ["B"],
"B": ["C", "F"],
"C": ["D"],
"D": ["E", "B"],
"E": [],
"F": ["G", "I"],
"G": ["H"],
"H": ["D"],
"I": ["H"],
},
code={
"A": dedent(
"""\
i = 1
"""
),
"B": dedent(
"""\
a = temp_0
c = temp_1
if a < c:
pass
"""
),
"C": dedent(
"""\
b = temp_2
c = temp_3
d = temp_4
"""
),
"D": dedent(
"""\
y = a + b
z = c + d
i = i + 1
if i < 100:
pass
"""
),
"E": "return\n",
"F": dedent(
"""\
a = temp_5
d = temp_6
if a < d:
pass
"""
),
"G": dedent(
"""\
d = temp
"""
),
"H": dedent(
"""\
b = temp
"""
),
"I": dedent(
"""\
c = temp
"""
),
},
block_type={"A": ParentScopeBlock},
)
cfg_real = Cfg()
cfg_real.block_list = blocks
cfg_real.as_tree = as_tree
cfg_real.root = cfg_real.block_list[0]
cfg_real.root.blocks = blocks
cfg_real.fill_df()
cfg_real.gather_initial_info()
cfg_real.compute_live_out()
cfg_real.ins_phi_function_all()
expected_phi_list = {
"A": set(),
"B": {"i"},
"C": set(),
"D": {"a", "b", "c", "d"},
"E": set(),
"F": set(),
"G": set(),
"H": {"c", "d"},
"I": set(),
}
self.assertPhiListEqual(cfg_real.block_list, expected_phi_list)