xref: /llvm-project/llvm/test/tools/llvm-reduce/Inputs/reduce-distinct-metadata.py (revision 42067f26cd084d29fdd08a75a36b8785ae9f3982)
1*42067f26SRobert Barinov# Helper script for distinct metadata reduction test
2*42067f26SRobert Barinov
3*42067f26SRobert Barinovimport sys
4*42067f26SRobert Barinovimport re
5*42067f26SRobert Barinov
6*42067f26SRobert Barinovinput = open(sys.argv[1], "r").read().splitlines()
7*42067f26SRobert Barinov
8*42067f26SRobert Barinovdepth_map = {"0": 1, "1": 3, "2": 3, "3": 2, "4": 1}
9*42067f26SRobert Barinov
10*42067f26SRobert Barinov
11*42067f26SRobert Barinovfor i in range(len(depth_map)):
12*42067f26SRobert Barinov    counter = 0
13*42067f26SRobert Barinov    for line in input:
14*42067f26SRobert Barinov        if re.match(rf".*interesting_{i}.*", line) != None:
15*42067f26SRobert Barinov            counter += 1
16*42067f26SRobert Barinov    if counter != depth_map[str(i)]:
17*42067f26SRobert Barinov        sys.exit(1)
18*42067f26SRobert Barinov
19*42067f26SRobert Barinovsys.exit(0)
20