def put_test()

in tasks/count-or-mem/generate_data.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def put_test(root, length_from, length_to):
    with open(f'{root}/test.src', 'w') as test_src, open(f'{root}/test.dst', 'w') as test_dst:
        for length in range(length_from, length_to):
            test_input = ' '.join(['a'] * length)
            print(test_input, file=test_src)

            test_output = ''  # test output is never used
            print(test_output, file=test_dst)

    shutil.copy(f'{root}/test.src', f'{root}/valid.src')
    shutil.copy(f'{root}/test.dst', f'{root}/valid.dst')