Commit 44f7d1d2 authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

using almost equal in all unittests

parent f10ddc71
...@@ -120,5 +120,5 @@ class TestSignaturizer(unittest.TestCase): ...@@ -120,5 +120,5 @@ class TestSignaturizer(unittest.TestCase):
res_all = module.predict(self.test_smiles) res_all = module.predict(self.test_smiles)
for idx, smile in enumerate(self.test_smiles): for idx, smile in enumerate(self.test_smiles):
res_single = module.predict([smile]) res_single = module.predict([smile])
self.assertTrue( np.testing.assert_almost_equal(res_all.signature[idx],
all(res_all.signature[idx] == res_single.signature[0])) res_single.signature[0])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment