import sys OK = 0 PE = 1 WA = 2 UH = 3 if len(sys.argv) < 4: print("Too few arguments for testing script") exit(UH) input_file = sys.argv[1] output_file = sys.argv[2] answer_file = sys.argv[3] try: with open(output_file) as f: output_data = list(map(int, f.read().split())) except: print("Wrong answer format") exit(PE) try: with open(answer_file) as f: answer_data = list(map(int, f.read().split())) except: print("Failed to parse answer file") exit(UH) if len(output_data) < len(answer_data): print('Too few numbers') exit(PE) if len(output_data) > len(answer_data): print('Extra numbers') exit(PE) for (n, (i, j)) in enumerate(zip(output_data, answer_data)): if i != j: print(f'{i} instead of {j} in pos {n}', end='') exit(0)