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)