File size: 2,885 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 | import os
from pathlib import Path
import pandas as pd
from src.tests.test_samples.sample2.boilerplate import (
CONFIG_BOILERPLATE,
TEMPLATE_BOILERPLATE,
)
from src.tests.utils import (
generate_write_jsons_and_run,
remove_file,
run_entry_point,
setup_mocker_patches,
)
FROZEN_TIMESTAMP = "1970-01-01"
CURRENT_DIR = Path("src/tests")
BASE_SAMPLE_PATH = CURRENT_DIR.joinpath("test_samples", "sample2")
BASE_RESULTS_CSV_PATH = os.path.join(
"outputs", BASE_SAMPLE_PATH, "Results", "Results_05AM.csv"
)
BASE_MULTIMARKED_CSV_PATH = os.path.join(
"outputs", BASE_SAMPLE_PATH, "Manual", "MultiMarkedFiles.csv"
)
def run_sample(mocker, input_path):
setup_mocker_patches(mocker)
output_dir = os.path.join("outputs", input_path)
run_entry_point(input_path, output_dir)
def extract_output_data(path):
output_data = pd.read_csv(path, keep_default_na=False)
return output_data
write_jsons_and_run = generate_write_jsons_and_run(
run_sample,
sample_path=BASE_SAMPLE_PATH,
template_boilerplate=TEMPLATE_BOILERPLATE,
config_boilerplate=CONFIG_BOILERPLATE,
)
def test_config_low_dimensions(mocker):
def modify_config(config):
config["dimensions"]["processing_height"] = 1000
config["dimensions"]["processing_width"] = 1000
exception = write_jsons_and_run(mocker, modify_config=modify_config)
assert str(exception) == "No Error"
def test_different_bubble_dimensions(mocker):
# Prevent appending to output csv:
remove_file(BASE_RESULTS_CSV_PATH)
remove_file(BASE_MULTIMARKED_CSV_PATH)
exception = write_jsons_and_run(mocker)
assert str(exception) == "No Error"
original_output_data = extract_output_data(BASE_RESULTS_CSV_PATH)
def modify_template(template):
# Incorrect global bubble size
template["bubbleDimensions"] = [5, 5]
# Correct bubble size for MCQBlock1a1
template["fieldBlocks"]["MCQBlock1a1"]["bubbleDimensions"] = [32, 32]
# Incorrect bubble size for MCQBlock1a11
template["fieldBlocks"]["MCQBlock1a11"]["bubbleDimensions"] = [10, 10]
remove_file(BASE_RESULTS_CSV_PATH)
remove_file(BASE_MULTIMARKED_CSV_PATH)
exception = write_jsons_and_run(mocker, modify_template=modify_template)
assert str(exception) == "No Error"
results_output_data = extract_output_data(BASE_RESULTS_CSV_PATH)
assert results_output_data.empty
output_data = extract_output_data(BASE_MULTIMARKED_CSV_PATH)
equal_columns = [f"q{i}" for i in range(1, 18)]
assert (
output_data[equal_columns].iloc[0].to_list()
== original_output_data[equal_columns].iloc[0].to_list()
)
unequal_columns = [f"q{i}" for i in range(168, 185)]
assert not (
output_data[unequal_columns].iloc[0].to_list()
== original_output_data[unequal_columns].iloc[0].to_list()
)
|