Commit f7bc7a2e authored by nd-02110114's avatar nd-02110114
Browse files

🐛 fix test

parent 175f1edf
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -13,6 +13,6 @@ class TestOneHotFeaturizer(TestCase):
    featurizer = dc.feat.one_hot.OneHotFeaturizer(dc.feat.one_hot.zinc_charset)
    one_hots = featurizer.featurize(mols)
    untransformed = featurizer.untransform(one_hots)
    len(smiles) == len(untransformed)
    assert len(smiles) == len(untransformed)
    for i in range(len(smiles)):
      assert smiles[i] == untransformed[i][0]