File size: 3,157 Bytes
b400ace | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | 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
|