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):
res_all = module.predict(self.test_smiles)
for idx, smile in enumerate(self.test_smiles):
res_single = module.predict([smile])
self.assertTrue(
all(res_all.signature[idx] == res_single.signature[0]))
np.testing.assert_almost_equal(res_all.signature[idx],
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