|
|
import pytest |
|
|
import pandas as pd |
|
|
from constants import MINIMAL_NUMBER_OF_ROWS, ASSAY_LIST, ANTIBODY_NAMES |
|
|
|
|
|
|
|
|
@pytest.fixture |
|
|
def valid_csv_data(): |
|
|
return { |
|
|
"antibody_name": ANTIBODY_NAMES[:MINIMAL_NUMBER_OF_ROWS], |
|
|
"vh_protein_sequence": [ |
|
|
"EVQLVESGGGLVQPGGSLRLSCAASGFTFSSYAMHWVRQAPGKGLEWVSAISGSGGSTYYADSVKGRFTISRDNSKNTLYLQMNSLRAEDTAVYYCARDYGDGYYFDYWGQGTLVTVSS" |
|
|
] |
|
|
* MINIMAL_NUMBER_OF_ROWS, |
|
|
"vl_protein_sequence": [ |
|
|
"DIQMTQSPSSLSASVGDRVTITCRASQSISSYLNWYQQKPGKAPKLLIYAASTLQSGVPSRFSGSGSGTDFTLTISSLQPEDFATYYCQQSYSTPFTFGQGTKVEIK" |
|
|
] |
|
|
* MINIMAL_NUMBER_OF_ROWS, |
|
|
**{assay: [0.85] * MINIMAL_NUMBER_OF_ROWS for assay in ASSAY_LIST}, |
|
|
} |
|
|
|
|
|
|
|
|
@pytest.fixture |
|
|
def valid_input_dataframe(valid_csv_data): |
|
|
return pd.DataFrame(valid_csv_data) |
|
|
|
|
|
|
|
|
@pytest.fixture |
|
|
def valid_csv_content(valid_input_dataframe): |
|
|
return valid_input_dataframe.to_csv(index=False) |
|
|
|
