# Produce file via
# $ git grep -n assertTrue | grep ' is ' > change_it.txt
# Manually made changes for lines with 12 space indent and for
# the files in the Speech package

import collections

def main():
    with open('change_it.txt', 'r') as file_obj:
        lines = file_obj.read().strip().split('\n')

    by_file = collections.defaultdict(dict)
    for line in lines:
        filename, line_no, content = line.split(':')
        line_no = int(line_no)
        in_file = by_file[filename]
        if line_no in in_file:
            raise KeyError('WAT', filename, line_no)
        if content[:24] != '        self.assertTrue(':
            raise ValueError(line, 'bad 1')
        if content[-1] != ')':
            raise ValueError(line, 'bad 2')
        left, right = content[24:-1].split(' is ')
        if ' ' in left or ' ' in right:
            raise ValueError(line, 'bad 3')

        actual = '        self.assertIs({}, {})'.format(left, right)
        in_file[line_no] = (content, actual)

    for filename, in_file in by_file.items():
        with open(filename, 'r') as file_obj:
            lines = file_obj.read().split('\n')

        for line_no, pair in in_file.items():
            assert lines[line_no - 1] == pair[0]
            lines[line_no - 1] = pair[1]

        with open(filename, 'w') as file_obj:
            file_obj.write('\n'.join(lines))


if __name__ == '__main__':
    main()