Merge pull request #144 from lucile-hashimoto/sulcal_graph

Debug test function for sulcal depth

Author: gauzias

tests/test_sulcal_depth.py

@@ -6,21 +6,23 @@
 
 acceptable_error = 0.0001
 
+
 class TestSulcalDepth(unittest.TestCase):
     test_mesh = sio.load_mesh('examples/data/example_mesh.gii')
     test_dpf = sio.load_texture('examples/data/example_dpf.gii').darray[0]
     test_dpf_star = sio.load_texture('examples/data/example_dpf_star.gii').darray[0]
-    dpf = sdepth.depth_potential_function(test_mesh)[0]
+    dpf = sdepth.depth_potential_function(test_mesh, alphas=[0.3])[0]
     dpf_star = sdepth.dpf_star(test_mesh)[0]
+
     def test_basic(self):
         # test size
-        self.assertTrue(self.test_dpf.shape==self.dpf.shape)
+        self.assertTrue(self.test_dpf.shape == self.dpf.shape)
         self.assertTrue(self.test_dpf_star.shape == self.dpf_star.shape)
 
     def test_consistency(self):
-        self.assertTrue(np.max(np.abs(self.test_dpf-self.dpf)) < acceptable_error)
+        self.assertTrue(np.max(np.abs(self.test_dpf - self.dpf)) < acceptable_error)
         self.assertTrue(np.max(np.abs(self.test_dpf_star - self.dpf_star)) < acceptable_error)
 
-if __name__ == "__main__":
 
+if __name__ == "__main__":
     unittest.main()

tests/test_sulcal_graph.py

@@ -1,7 +1,7 @@
 import unittest
 import numpy as np
 import networkx as nx
-from slam import sulcal_graph
+from slam import sulcal_graph, geodesics
 from slam import texture
 
 
@@ -65,8 +65,9 @@ def test_add_geodesic_distances(self):
         def mock_compute_gdist(mesh, ridge):
             return np.array([0.0, 1.0, 2.0, 3.0, 4.0])
 
-        sulcal_graph.geodesics.compute_gdist = mock_compute_gdist
+        geodesics.compute_gdist = mock_compute_gdist
         g = sulcal_graph.add_geodesic_distances_to_graph(self.graph, self.mesh, save=False)
+        print(g.edges[list(g.edges)[0]])
 
         self.assertIn("geodesic_distance_btw_ridge_pit_i", g.edges[list(g.edges)[0]].keys())
         self.assertIn("geodesic_distance_btw_ridge_pit_j", g.edges[list(g.edges)[0]].keys())