def put_train_fpa()

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

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

def put_train_fpa(root, train_N):
    alphabet_input = string.ascii_lowercase[:train_N]
    alphabet_output = string.ascii_uppercase[:train_N] 

    with open(f'{root}/train.src', 'w') as train_src, open(f'{root}/train.dst', 'w') as train_tgt:
        # Primitives
        for train_input, train_output in zip(alphabet_input, alphabet_output):
            print(train_input, file=train_src)
            print(train_output, file=train_tgt)

        # Function
        train_input = 'F ' + alphabet_input[0]
        print(train_input, file=train_src)
        train_output = ' '.join([alphabet_output[0]] * 3)
        print(train_output, file=train_tgt)