@@ -77,7 +77,7 @@ def test_sklearn_pipeline(self) -> None:
7777 )
7878 s_pipeline .fit (TEST_SMILES , CONTAINS_OX )
7979 predicted_value_array = s_pipeline .predict (TEST_SMILES )
80- for pred_val , true_val in zip (predicted_value_array , CONTAINS_OX , strict = False ):
80+ for pred_val , true_val in zip (predicted_value_array , CONTAINS_OX , strict = True ):
8181 self .assertEqual (pred_val , true_val )
8282
8383 def test_sklearn_pipeline_parallel (self ) -> None :
@@ -96,7 +96,7 @@ def test_sklearn_pipeline_parallel(self) -> None:
9696 s_pipeline .fit (TEST_SMILES , CONTAINS_OX )
9797 out = s_pipeline .predict (TEST_SMILES )
9898 self .assertEqual (len (out ), len (CONTAINS_OX ))
99- for pred_val , true_val in zip (out , CONTAINS_OX , strict = False ):
99+ for pred_val , true_val in zip (out , CONTAINS_OX , strict = True ):
100100 self .assertEqual (pred_val , true_val )
101101
102102 def test_salt_removal (self ) -> None :
@@ -125,7 +125,7 @@ def test_salt_removal(self) -> None:
125125 for generated_smiles , smiles_without_salt in zip (
126126 generated_smiles_list ,
127127 smiles_without_salt_list ,
128- strict = False ,
128+ strict = True ,
129129 ):
130130 self .assertEqual (generated_smiles , smiles_without_salt )
131131
@@ -160,7 +160,7 @@ def test_json_generation(self) -> None:
160160 for loaded_element , original_element in zip (
161161 loaded_pipeline .steps ,
162162 pipeline_element_list ,
163- strict = False ,
163+ strict = True ,
164164 ):
165165 if loaded_element [1 ] == "passthrough" :
166166 self .assertEqual (loaded_element [1 ], original_element )
0 commit comments