Skip to content

Commit d898d5e

Browse files
change zip's strict to True
1 parent 75cb6a1 commit d898d5e

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tests/test_pipeline.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)