setup_test_files.py

import os.path
import shutil
dir = os.path.dirname(__file__)


def setup(source, dest):
    if os.path.isdir(dest):
        shutil.rmtree(dest)
    os.mkdir(dest)
    os.mkdir(dest + "/output")
    from snipper import snip_dir
    snip_dir(source, dest_dir=dest + "/code", clean_destination_dir=True, output_dir=dest + "/output")


def setup_keep(source, dest):
    from snipper import snipper_main
    # snip_dir(source, dest_dir=dest + "/code", clean_destination_dir=True, output_dir=dest + "/output")
    with open(source, 'r') as f:
        lines = f.read().strip().splitlines()

    lines2 = snipper_main.censor_code(lines, keep=True)
    assert len(lines) == len(lines2)
    with open(dest, 'w') as f:
        f.write("\n".join(lines2))
    return lines, lines2

if __name__ == "__main__":
    print("File used for setting up sets -- don't run this script unless repository known to work (Tue, 2022)")
    # assert False
    # if os.path.isdir("student_repo"):
    #     shutil.rmtree("student_repo")

    # from snipper import snip_dir
    # if not os.path.isdir("student_correct_output"):
    #     os.mkdir("student_correct_output")
    for file in ['demo1']:
        setup(dir+"/" + file, dir+"/" + file+"_correct")
    setup_keep(dir+"/demo2/framework.py", dir+"/demo2/framework_correct.txt")

    a= 234
    # assert False