def put_test()

in tasks/dax-compo/generate_data.py [0:0]

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

def put_test(root, train_N):

    alphabet_input = string.ascii_lowercase[:train_N]
    alphabet_output = string.ascii_uppercase[:train_N] 

    with open(f'{root}/test.src', 'w') as test_src, open(f'{root}/test.dst', 'w') as test_dst:
        test_output = '' # test output is never used

        for char_inp in alphabet_input:
            test_input = char_inp
            print(test_input, file=test_src)
            print(test_output, file=test_dst)

            test_input = 'F '+ char_inp
            print(test_input, file=test_src)
            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')