Merge pull request #3662 from shreyasvinaya/CI-Fix

CI Unit tests fix
deepchem · Nov 8, 2023 · 6de1881 · 6de1881
2 parents 933e3b9 + b586e7e
commit 6de1881
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/deepchem/data/tests/test_fasta_loader.py b/deepchem/data/tests/test_fasta_loader.py
@@ -45,7 +45,7 @@ def test_fasta_one_hot_big(self):
         ]
         input_file = os.path.join(self.current_dir, "uniprot_truncated.fasta")
         loader = dc.data.FASTALoader(OneHotFeaturizer(charset=protein,
-                                                      max_length=1000),
+                                                      max_length=1165),
                                      legacy=False)
         sequences = loader.create_dataset(input_file)
 

diff --git a/deepchem/feat/tests/test_dmpnn_featurizer.py b/deepchem/feat/tests/test_dmpnn_featurizer.py
@@ -97,7 +97,8 @@ def test_featurizer_ring(test_parameters):
     if use_original_atom_ranks:
         assert (graph_feat[0].edge_index == edge_index_orignal_order[0]).all()
     else:
-        if graph_feat[0].edge_index.shape == edge_index_orignal_order[0]:
+        if np.array_equal(graph_feat[0].edge_index.shape,
+                          edge_index_orignal_order[0]):
             assert (graph_feat[0].edge_index !=
                     edge_index_orignal_order[0]).any()
 
@@ -152,7 +153,8 @@ def test_featurizer_general_case(test_parameters):
     if use_original_atom_ranks:
         assert (graph_feat[1].edge_index == edge_index_orignal_order[1]).all()
     else:
-        if graph_feat[1].edge_index.shape == edge_index_orignal_order[1]:
+        if np.array_equal(graph_feat[1].edge_index.shape,
+                          edge_index_orignal_order[1]):
             assert (graph_feat[1].edge_index !=
                     edge_index_orignal_order[1]).any()
 
@@ -206,7 +208,8 @@ def test_featurizer_single_atom(test_parameters):
     if use_original_atom_ranks:
         assert (graph_feat[2].edge_index == edge_index_orignal_order[2]).all()
     else:
-        if graph_feat[2].edge_index.shape == edge_index_orignal_order[2]:
+        if np.array_equal(graph_feat[2].edge_index.shape,
+                          edge_index_orignal_order[2]):
             # the atom order for 'C' is same in case of canonical and original ordering
             assert (
                 graph_feat[2].edge_index == edge_index_orignal_order[2]).all()