| # ===----------------------------------------------------------------------===## |
| # |
| # Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| # |
| # ===----------------------------------------------------------------------===## |
| import unittest |
| |
| import filecheck_lint as fcl |
| |
| |
| class TestParser(unittest.TestCase): |
| |
| def test_parse_all_additional_prefixes(self): |
| |
| def run(content, expected_prefixes): |
| prefixes = set(fcl.parse_custom_prefixes(content)) |
| for prefix in expected_prefixes: |
| self.assertIn(prefix, prefixes) |
| |
| for content, expected_prefixes in [ |
| ('-check-prefix=PREFIX', {'PREFIX'}), |
| ('-check-prefix=\'PREFIX\'', {'PREFIX'}), |
| ('-check-prefix="PREFIX"', {'PREFIX'}), |
| ('-check-prefix PREFIX', {'PREFIX'}), |
| ('-check-prefix PREFIX', {'PREFIX'}), |
| ('-check-prefixes=PREFIX1,PREFIX2', {'PREFIX1', 'PREFIX2'}), |
| ('-check-prefixes PREFIX1,PREFIX2', {'PREFIX1', 'PREFIX2'}), |
| ( |
| """-check-prefix=PREFIX1 -check-prefix PREFIX2 |
| -check-prefixes=PREFIX3,PREFIX4 -check-prefix=PREFIX5 |
| -check-prefixes PREFIX6,PREFIX7 -check-prefixes=PREFIX8', |
| """, # pylint: disable=bad-continuation |
| {f'PREFIX{i}' for i in range(1, 9)}), |
| ]: |
| run(content, expected_prefixes) |
| |
| def test_additional_prefixes_uniquely(self): |
| lines = ['--check-prefix=SOME-PREFIX', '--check-prefix=SOME-PREFIX'] |
| prefixes = set(fcl.parse_custom_prefixes('\n'.join(lines))) |
| assert len(prefixes) == 1 |
| |
| |
| class TestTypoDetection(unittest.TestCase): |
| |
| def test_find_potential_directives_comment_prefix(self): |
| lines = ['junk; CHCK1:', 'junk// CHCK2:', 'SOME CHCK3:'] |
| content = '\n'.join(lines) |
| |
| results = list(fcl.find_potential_directives(content)) |
| assert len(results) == 3 |
| pos, match = results[0] |
| assert (pos.line == 1 and |
| pos.start_column == len('junk; ') + 1 and |
| pos.end_column == len(lines[0]) - 1) |
| assert match == 'CHCK1' |
| |
| pos, match = results[1] |
| assert (pos.line == 2 and |
| pos.start_column == len('junk// ') + 1 and |
| pos.end_column == len(lines[1]) - 1) |
| assert match == 'CHCK2' |
| |
| pos, match = results[2] |
| assert (pos.line == 3 and |
| pos.start_column == 1 and |
| pos.end_column == len(lines[2]) - 1) |
| assert match == 'SOME CHCK3' |
| |
| def test_levenshtein(self): |
| for s1, s2, distance in [ |
| ('Levenshtein', 'Levenstin', 2), # 2 insertions |
| ('Levenshtein', 'Levenstherin', 3), # 1 insertion, 2 deletions |
| ('Levenshtein', 'Lenvinshtein', 2), # 1 deletion, 1 substitution |
| ('Levenshtein', 'Levenshtein', 0), # identical strings |
| ]: |
| assert fcl.levenshtein(s1, s2) == distance |