| import os | |
| import shutil | |
| from glob import glob | |
| from src.tests.utils import run_entry_point, setup_mocker_patches | |
| def read_file(path): | |
| with open(path) as file: | |
| return file.read() | |
| def run_sample(mocker, sample_path): | |
| setup_mocker_patches(mocker) | |
| input_path = os.path.join("samples", sample_path) | |
| output_dir = os.path.join("outputs", sample_path) | |
| if os.path.exists(output_dir): | |
| print( | |
| f"Warning: output directory already exists: {output_dir}. This may affect the test execution." | |
| ) | |
| run_entry_point(input_path, output_dir) | |
| sample_outputs = extract_sample_outputs(output_dir) | |
| print(f"Note: removing output directory: {output_dir}") | |
| shutil.rmtree(output_dir) | |
| return sample_outputs | |
| EXT = "*.csv" | |
| def extract_sample_outputs(output_dir): | |
| sample_outputs = {} | |
| for _dir, _subdir, _files in os.walk(output_dir): | |
| for file in glob(os.path.join(_dir, EXT)): | |
| relative_path = os.path.relpath(file, output_dir) | |
| sample_outputs[relative_path] = read_file(file) | |
| return sample_outputs | |
| def test_run_answer_key_using_csv(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "answer-key/using-csv") | |
| assert snapshot == sample_outputs | |
| def test_run_answer_key_weighted_answers(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "answer-key/weighted-answers") | |
| assert snapshot == sample_outputs | |
| def test_run_sample1(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "sample1") | |
| assert snapshot == sample_outputs | |
| def test_run_sample2(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "sample2") | |
| assert snapshot == sample_outputs | |
| def test_run_sample3(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "sample3") | |
| assert snapshot == sample_outputs | |
| def test_run_sample4(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "sample4") | |
| assert snapshot == sample_outputs | |
| def test_run_sample5(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "sample5") | |
| assert snapshot == sample_outputs | |
| def test_run_sample6(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "sample6") | |
| assert snapshot == sample_outputs | |
| def test_run_community_Antibodyy(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "community/Antibodyy") | |
| assert snapshot == sample_outputs | |
| def test_run_community_ibrahimkilic(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "community/ibrahimkilic") | |
| assert snapshot == sample_outputs | |
| def test_run_community_Sandeep_1507(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "community/Sandeep-1507") | |
| assert snapshot == sample_outputs | |
| def test_run_community_Shamanth(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "community/Shamanth") | |
| assert snapshot == sample_outputs | |
| def test_run_community_UmarFarootAPS(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "community/UmarFarootAPS") | |
| assert snapshot == sample_outputs | |
| def test_run_community_UPSC_mock(mocker, snapshot): | |
| sample_outputs = run_sample(mocker, "community/UPSC-mock") | |
| assert snapshot == sample_outputs | |