| # Copyright 2019 Google LLC |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # https://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| """Tests for dependency_checker.py.""" |
| |
| import unittest |
| from compiler.front_end import dependency_checker |
| from compiler.front_end import glue |
| from compiler.front_end import test_util |
| from compiler.util import error |
| |
| |
| def _parse_snippet(emb_file): |
| ir, unused_debug_info, errors = glue.parse_emboss_file( |
| "m.emb", |
| test_util.dict_file_reader({"m.emb": emb_file}), |
| stop_before_step="find_dependency_cycles") |
| assert not errors |
| return ir |
| |
| |
| def _find_dependencies_for_snippet(emb_file): |
| ir, unused_debug_info, errors = glue.parse_emboss_file( |
| "m.emb", |
| test_util.dict_file_reader({ |
| "m.emb": emb_file |
| }), |
| stop_before_step="set_dependency_order") |
| assert not errors, errors |
| return ir |
| |
| |
| class DependencyCheckerTest(unittest.TestCase): |
| |
| def test_error_on_simple_field_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " 0 [+field2] UInt field1\n" |
| " 0 [+field1] UInt field2\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nfield1"), |
| error.note("m.emb", struct.field[1].source_location, "field2") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_self_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " 0 [+field1] UInt field1\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nfield1") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_triple_field_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " 0 [+field2] UInt field1\n" |
| " 0 [+field3] UInt field2\n" |
| " 0 [+field1] UInt field3\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nfield1"), |
| error.note("m.emb", struct.field[1].source_location, "field2"), |
| error.note("m.emb", struct.field[2].source_location, "field3"), |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_complex_field_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " 0 [+field2] UInt field1\n" |
| " 0 [+field3+field4] UInt field2\n" |
| " 0 [+field1] UInt field3\n" |
| " 0 [+field2] UInt field4\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nfield1"), |
| error.note("m.emb", struct.field[1].source_location, "field2"), |
| error.note("m.emb", struct.field[2].source_location, "field3"), |
| error.note("m.emb", struct.field[3].source_location, "field4"), |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_simple_enum_value_cycle(self): |
| ir = _parse_snippet("enum Foo:\n" |
| " XX = YY\n" |
| " YY = XX\n") |
| enum = ir.module[0].type[0].enumeration |
| self.assertEqual([[ |
| error.error("m.emb", enum.value[0].source_location, |
| "Dependency cycle\nXX"), |
| error.note("m.emb", enum.value[1].source_location, "YY") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_no_error_on_no_cycle(self): |
| ir = _parse_snippet("enum Foo:\n" |
| " XX = 0\n" |
| " YY = XX\n") |
| self.assertEqual([], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_cycle_nested(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " struct Bar:\n" |
| " 0 [+field2] UInt field1\n" |
| " 0 [+field1] UInt field2\n" |
| " 0 [+1] UInt field\n") |
| struct = ir.module[0].type[0].subtype[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nfield1"), |
| error.note("m.emb", struct.field[1].source_location, "field2") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_import_cycle(self): |
| ir, unused_debug_info, errors = glue.parse_emboss_file( |
| "m.emb", |
| test_util.dict_file_reader({"m.emb": 'import "n.emb" as n\n', |
| "n.emb": 'import "m.emb" as m\n'}), |
| stop_before_step="find_dependency_cycles") |
| assert not errors |
| self.assertEqual([[ |
| error.error("m.emb", ir.module[0].source_location, |
| "Import dependency cycle\nm.emb"), |
| error.note("n.emb", ir.module[2].source_location, "n.emb") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_import_cycle_and_field_cycle(self): |
| ir, unused_debug_info, errors = glue.parse_emboss_file( |
| "m.emb", |
| test_util.dict_file_reader({"m.emb": 'import "n.emb" as n\n' |
| "struct Foo:\n" |
| " 0 [+field1] UInt field1\n", |
| "n.emb": 'import "m.emb" as m\n'}), |
| stop_before_step="find_dependency_cycles") |
| assert not errors |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", ir.module[0].source_location, |
| "Import dependency cycle\nm.emb"), |
| error.note("n.emb", ir.module[2].source_location, "n.emb") |
| ], [ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nfield1") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_field_existence_self_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " if x == 1:\n" |
| " 0 [+1] UInt x\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nx") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_field_existence_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " if y == 1:\n" |
| " 0 [+1] UInt x\n" |
| " if x == 0:\n" |
| " 1 [+1] UInt y\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nx"), |
| error.note("m.emb", struct.field[1].source_location, "y") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_virtual_field_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " let x = y\n" |
| " let y = x\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nx"), |
| error.note("m.emb", struct.field[1].source_location, "y") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_virtual_non_virtual_field_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " let x = y\n" |
| " x [+4] UInt y\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nx"), |
| error.note("m.emb", struct.field[1].source_location, "y") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_non_virtual_virtual_field_cycle(self): |
| ir = _parse_snippet("struct Foo:\n" |
| " y [+4] UInt x\n" |
| " let y = x\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nx"), |
| error.note("m.emb", struct.field[1].source_location, "y") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_error_on_cycle_involving_subfield(self): |
| ir = _parse_snippet("struct Bar:\n" |
| " foo_b.x [+4] Foo foo_a\n" |
| " foo_a.x [+4] Foo foo_b\n" |
| "struct Foo:\n" |
| " 0 [+4] UInt x\n") |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([[ |
| error.error("m.emb", struct.field[0].source_location, |
| "Dependency cycle\nfoo_a"), |
| error.note("m.emb", struct.field[1].source_location, "foo_b") |
| ]], dependency_checker.find_dependency_cycles(ir)) |
| |
| def test_dependency_ordering_with_no_dependencies(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " 0 [+4] UInt a\n" |
| " 4 [+4] UInt b\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([0, 1], struct.fields_in_dependency_order[:2]) |
| |
| def test_dependency_ordering_with_dependency_in_order(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " 0 [+4] UInt a\n" |
| " a [+4] UInt b\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([0, 1], struct.fields_in_dependency_order[:2]) |
| |
| def test_dependency_ordering_with_dependency_in_reverse_order(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " b [+4] UInt a\n" |
| " 0 [+4] UInt b\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([1, 0], struct.fields_in_dependency_order[:2]) |
| |
| def test_dependency_ordering_with_extra_fields(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " d [+4] UInt a\n" |
| " 4 [+4] UInt b\n" |
| " 8 [+4] UInt c\n" |
| " 12 [+4] UInt d\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([1, 2, 3, 0], struct.fields_in_dependency_order[:4]) |
| |
| def test_dependency_ordering_scrambled(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " d [+4] UInt a\n" |
| " c [+4] UInt b\n" |
| " a [+4] UInt c\n" |
| " 12 [+4] UInt d\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([3, 0, 2, 1], struct.fields_in_dependency_order[:4]) |
| |
| def test_dependency_ordering_multiple_dependents(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " d [+4] UInt a\n" |
| " d [+4] UInt b\n" |
| " d [+4] UInt c\n" |
| " 12 [+4] UInt d\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([3, 0, 1, 2], struct.fields_in_dependency_order[:4]) |
| |
| def test_dependency_ordering_multiple_dependencies(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " b+c [+4] UInt a\n" |
| " 4 [+4] UInt b\n" |
| " 8 [+4] UInt c\n" |
| " a [+4] UInt d\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([1, 2, 0, 3], struct.fields_in_dependency_order[:4]) |
| |
| def test_dependency_ordering_with_parameter(self): |
| ir = _find_dependencies_for_snippet("struct Foo:\n" |
| " 0 [+1] Bar(x) b\n" |
| " 1 [+1] UInt x\n" |
| "struct Bar(x: UInt:8):\n" |
| " x [+1] UInt y\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([1, 0], struct.fields_in_dependency_order[:2]) |
| |
| def test_dependency_ordering_with_local_parameter(self): |
| ir = _find_dependencies_for_snippet("struct Foo(x: Int:13):\n" |
| " 0 [+x] Int b\n") |
| self.assertEqual([], dependency_checker.set_dependency_order(ir)) |
| struct = ir.module[0].type[0].structure |
| self.assertEqual([0], struct.fields_in_dependency_order[:1]) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |
