update data paths in test_dataset.py

DeepRank · Oct 21, 2023 · 0a068b5 · 0a068b5
1 parent fc5f6af
commit 0a068b5
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
@@ -985,15 +985,15 @@ def test_incompatible_dataset_train_type(self):
 
     def test_invalid_pretrained_model_path(self):
 
-        hdf5_graph = "tests/data/hdf5/train.hdf5"
+        hdf5_graph = "tests/data/hdf5/test.hdf5"
         with self.assertRaises(ValueError):
             GraphDataset(
                 hdf5_path = hdf5_graph,
                 train = False,
                 train_data = hdf5_graph
             )
 
-        hdf5_grid = "tests/data/hdf5/grid_data.hdf5"
+        hdf5_grid = "tests/data/hdf5/1ATN_ppi.hdf5"
         with self.assertRaises(ValueError):
             GridDataset(
                 hdf5_path = hdf5_grid,
@@ -1004,16 +1004,16 @@ def test_invalid_pretrained_model_path(self):
     def test_invalid_pretrained_model_data_type(self):
 
         hdf5_graph = "tests/data/hdf5/test.hdf5"
-        pretrained_grid_model = "tests/data/testing_grid_model.pth.tar"
+        pretrained_grid_model = "tests/data/pretrained/testing_grid_model.pth.tar"
         with self.assertRaises(TypeError):
             GraphDataset(
                 hdf5_path = hdf5_graph,
                 train = False,
                 train_data = pretrained_grid_model
             )
 
-        hdf5_grid = "tests/data/hdf5/grid_data.hdf5"
-        pretrained_graph_model = "tests/data/testing_graph_model.pth.tar"
+        hdf5_grid = "tests/data/hdf5/1ATN_ppi.hdf5"
+        pretrained_graph_model = "tests/data/pretrained/testing_graph_model.pth.tar"
         with self.assertRaises(TypeError):
             GridDataset(
                 hdf5_path = hdf5_grid,