@@ -20,31 +20,26 @@ def test_mol_to_smiles(mols_list):
2020 assert all (isinstance (x , str ) for x in smiles_list )
2121
2222
23- def test_mol_to_and_from_smiles (smiles_list ):
23+ def test_mol_to_and_from_smiles (mols_list ):
2424 mol_from_smiles = MolFromSmilesTransformer ()
2525 mol_to_smiles = MolToSmilesTransformer ()
2626
27- mols_list = mol_from_smiles .transform (smiles_list )
28- smiles_list_2 = mol_to_smiles .transform (mols_list )
29-
30- assert smiles_list_2 == smiles_list
27+ smiles_list_1 = mol_to_smiles .transform (mols_list )
28+ smiles_list_2 = mol_to_smiles .transform (
29+ mol_from_smiles . transform ( mol_to_smiles . transform ( mols_list ))
30+ )
3131
32+ assert smiles_list_1 == smiles_list_2
3233
33- def test_parallel_to_and_from_smiles (smiles_list ):
34- mol_from_smiles_seq = MolFromSmilesTransformer ()
35- mol_from_smiles_parallel = MolFromSmilesTransformer (n_jobs = - 1 )
3634
35+ def test_parallel_to_and_from_smiles (mols_list ):
3736 mol_to_smiles_seq = MolToSmilesTransformer ()
3837 mol_to_smiles_parallel = MolToSmilesTransformer (n_jobs = - 1 )
3938
40- mols_list_seq = mol_from_smiles_seq .transform (smiles_list )
41- mols_list_parallel = mol_from_smiles_parallel .transform (smiles_list )
42-
43- smiles_list_2_seq = mol_to_smiles_seq .transform (mols_list_seq )
44- smiles_list_2_parallel = mol_to_smiles_parallel .transform (mols_list_parallel )
39+ smiles_list_seq = mol_to_smiles_seq .transform (mols_list )
40+ smiles_list_parallel = mol_to_smiles_parallel .transform (mols_list )
4541
46- assert smiles_list_2_seq == smiles_list
47- assert smiles_list_2_seq == smiles_list_2_parallel
42+ assert smiles_list_seq == smiles_list_parallel
4843
4944
5045def test_from_invalid_smiles (smiles_list ):
0 commit comments