rm minimal number of rows from tests
Browse files- test/test_validation.py +2 -3
test/test_validation.py
CHANGED
@@ -2,7 +2,7 @@ import pytest
|
|
2 |
import pandas as pd
|
3 |
import gradio as gr
|
4 |
from validation import validate_csv_file, validate_csv_can_be_read, validate_dataframe
|
5 |
-
from constants import REQUIRED_COLUMNS,
|
6 |
|
7 |
|
8 |
class TestValidateCsvCanBeRead:
|
@@ -33,11 +33,10 @@ class TestValidateCsvCanBeRead:
|
|
33 |
# Create CSV with quoted fields and enough rows
|
34 |
base_row = 'test_antibody,"EVQLVESGGGLVQPGGSLRLSCAASGFTFSSYAMHWVRQAPGKGLEWVSAISGSGGSTYYADSVKGRFTISRDNSKNTLYLQMNSLRAEDTAVYYCARDYGDGYYFDYWGQGTLVTVSS","DIQMTQSPSSLSASVGDRVTITCRASQSISSYLNWYQQKPGKAPKLLIYAASTLQSGVPSRFSGSGSGTDFTLTISSLQPEDFATYYCQQSYSTPFTFGQGTKVEIK",95.2,0.85,0.92,0.78,0.81,72.5'
|
35 |
csv_content = "antibody_name,vh_protein_sequence,vl_protein_sequence,SEC %Monomer,HIC,PR_CHO,AC-SINS_pH6.0,AC-SINS_pH7.4,Tm\n"
|
36 |
-
csv_content += "\n".join([base_row] *
|
37 |
|
38 |
df = validate_csv_can_be_read(csv_content)
|
39 |
assert isinstance(df, pd.DataFrame)
|
40 |
-
assert len(df) == MINIMAL_NUMBER_OF_ROWS
|
41 |
|
42 |
|
43 |
class TestValidateDataframe:
|
|
|
2 |
import pandas as pd
|
3 |
import gradio as gr
|
4 |
from validation import validate_csv_file, validate_csv_can_be_read, validate_dataframe
|
5 |
+
from constants import REQUIRED_COLUMNS, ASSAY_LIST
|
6 |
|
7 |
|
8 |
class TestValidateCsvCanBeRead:
|
|
|
33 |
# Create CSV with quoted fields and enough rows
|
34 |
base_row = 'test_antibody,"EVQLVESGGGLVQPGGSLRLSCAASGFTFSSYAMHWVRQAPGKGLEWVSAISGSGGSTYYADSVKGRFTISRDNSKNTLYLQMNSLRAEDTAVYYCARDYGDGYYFDYWGQGTLVTVSS","DIQMTQSPSSLSASVGDRVTITCRASQSISSYLNWYQQKPGKAPKLLIYAASTLQSGVPSRFSGSGSGTDFTLTISSLQPEDFATYYCQQSYSTPFTFGQGTKVEIK",95.2,0.85,0.92,0.78,0.81,72.5'
|
35 |
csv_content = "antibody_name,vh_protein_sequence,vl_protein_sequence,SEC %Monomer,HIC,PR_CHO,AC-SINS_pH6.0,AC-SINS_pH7.4,Tm\n"
|
36 |
+
csv_content += "\n".join([base_row] * 10)
|
37 |
|
38 |
df = validate_csv_can_be_read(csv_content)
|
39 |
assert isinstance(df, pd.DataFrame)
|
|
|
40 |
|
41 |
|
42 |
class TestValidateDataframe:
|