Loading deepchem/feat/tests/test_mat_featurizer.py +3 −2 Original line number Diff line number Diff line Loading @@ -24,10 +24,11 @@ class TestMATFeaturizer(unittest.TestCase): out = featurizer.featurize(self.mol) assert (type(out) == np.ndarray) assert (out.shape == (1, 2, 31)) assert (out == np.array([[[ correct_array = np.array([[[ 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1. ], [ 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0., 1., 0. ]]])) ]]]) assert (np.array_equal(out, correct_array)) Loading
deepchem/feat/tests/test_mat_featurizer.py +3 −2 Original line number Diff line number Diff line Loading @@ -24,10 +24,11 @@ class TestMATFeaturizer(unittest.TestCase): out = featurizer.featurize(self.mol) assert (type(out) == np.ndarray) assert (out.shape == (1, 2, 31)) assert (out == np.array([[[ correct_array = np.array([[[ 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1. ], [ 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0., 1., 0. ]]])) ]]]) assert (np.array_equal(out, correct_array))
