pquintero commited on
Commit
21e1c3f
·
1 Parent(s): 5d5df93

rm minimal number of rows from tests

Browse files
Files changed (1) hide show
  1. 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, MINIMAL_NUMBER_OF_ROWS, ASSAY_LIST
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] * MINIMAL_NUMBER_OF_ROWS)
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: