def test_save(self, mock_writegeo):
""" Test the save method.
"""
# Test execution
surf = TriSurface(**self.kwargs)
out_file = "/my/path/mock_destination"
surf.save(out_file)
def test_save(self, mock_writegeo):
""" Test the save method.
"""
# Test execution
surf = TriSurface(**self.kwargs)
out_file = "/my/path/mock_destination"
surf.save(out_file)
def test_labelize(self):
""" Test the labelize method.
"""
# Test execution
surf = TriSurface(**self.kwargs)
surf.vertices += 1
label_array, nb_of_labels = surf.labelize(shape=(3, 3, 3))
self.assertEqual(6, nb_of_labels)
def test_savevtk(self, mock_vtkpoint, mock_vtkcell, mock_vtkuarray,
mock_vtktriangle, mock_vtkpoly, mock_vtkwrite):
""" Test the save vtk method.
"""
# Test execution
surf = TriSurface(**self.kwargs)
out_file = "/my/path/mock_destination"
surf.save_vtk(out_file, inflated=True)
def test_labelize(self):
""" Test the labelize method.
"""
# Test execution
surf = TriSurface(**self.kwargs)
surf.vertices += 1
label_array, nb_of_labels = surf.labelize(shape=(3, 3, 3))
self.assertEqual(6, nb_of_labels)
def test_savevtk(self, mock_vtkpoint, mock_vtkcell, mock_vtkuarray,
mock_vtktriangle, mock_vtkpoly, mock_vtkwrite):
""" Test the save vtk method.
"""
# Test execution
surf = TriSurface(**self.kwargs)
out_file = "/my/path/mock_destination"
surf.save_vtk(out_file, inflated=True)
def test_normal_execution(self):
""" Test the normal behaviour of the function.
"""
# Test execution
surf = TriSurface(**self.kwargs)
self.assertTrue(numpy.allclose(self.kwargs["vertices"], surf.vertices))
self.assertTrue(
numpy.allclose(self.kwargs["triangles"], surf.triangles))
self.assertTrue(
numpy.allclose(self.kwargs["inflated_vertices"],
surf.inflated_vertices))
self.assertEqual(surf.shape(), (6, 18, 8))
def test_normal_execution(self):
""" Test the normal behaviour of the function.
"""
# Test execution
surf = TriSurface(**self.kwargs)
self.assertTrue(numpy.allclose(self.kwargs["vertices"],
surf.vertices))
self.assertTrue(numpy.allclose(self.kwargs["triangles"],
surf.triangles))
self.assertTrue(numpy.allclose(self.kwargs["inflated_vertices"],
surf.inflated_vertices))
self.assertEqual(surf.shape(), (6, 18, 8))
def test_voxelize(self, mock_vtkpoint, mock_vtkpoly, mock_poly,
mock_vtkslect, mock_vtkutil, mock_vtkwrite):
""" Test the voxelize vtk method.
"""
# Set the mocked functions returned values
mock_poly.return_value = object()
mock_vtkutil.numpy_support.vtk_to_numpy.return_value = numpy.ones(
(3, 3, 3), dtype=int)
# Test execution
surf = TriSurface(**self.kwargs)
out_file = "/my/path/mock_destination"
inside_array = surf.voxelize(shape=(3, 3, 3), tol=0)
self.assertTrue(numpy.allclose(
inside_array,
mock_vtkutil.numpy_support.vtk_to_numpy.return_value))
def test_voxelize(self, mock_vtkpoint, mock_vtkpoly, mock_poly,
mock_vtkslect, mock_vtkutil, mock_vtkwrite):
""" Test the voxelize vtk method.
"""
# Set the mocked functions returned values
mock_poly.return_value = object()
mock_vtkutil.numpy_support.vtk_to_numpy.return_value = numpy.ones(
(3, 3, 3), dtype=int)
# Test execution
surf = TriSurface(**self.kwargs)
out_file = "/my/path/mock_destination"
inside_array = surf.voxelize(shape=(3, 3, 3), tol=0)
self.assertTrue(
numpy.allclose(
inside_array,
mock_vtkutil.numpy_support.vtk_to_numpy.return_value))
def test_load(self, mock_readgeo, mock_readannot):
""" Test the load method.
"""
# Set the mocked functions returned values
mock_readgeo.side_effect = [
(self.kwargs["vertices"], self.kwargs["triangles"]),
(self.kwargs["inflated_vertices"], self.kwargs["triangles"])]
mock_readannot.return_value = [
self.labels, self.ctab, self.labels]
# Test execution
meshfile = "/my/path/mock_mesh"
inflatedmeshpath = "/my/path/mock_infmesh"
annotfile = "/my/path/mock_annot"
surf = TriSurface.load(meshfile, inflatedmeshpath=inflatedmeshpath,
annotfile=annotfile)
self.assertTrue(numpy.allclose(self.kwargs["vertices"],
surf.vertices))
self.assertTrue(numpy.allclose(self.kwargs["triangles"],
surf.triangles))
self.assertTrue(numpy.allclose(self.kwargs["inflated_vertices"],
surf.inflated_vertices))
self.assertEqual(surf.shape(), (6, 18, 8))
def test_load(self, mock_readgeo, mock_readannot):
""" Test the load method.
"""
# Set the mocked functions returned values
mock_readgeo.side_effect = [
(self.kwargs["vertices"], self.kwargs["triangles"]),
(self.kwargs["inflated_vertices"], self.kwargs["triangles"])
]
mock_readannot.return_value = [self.labels, self.ctab, self.labels]
# Test execution
meshfile = "/my/path/mock_mesh"
inflatedmeshpath = "/my/path/mock_infmesh"
annotfile = "/my/path/mock_annot"
surf = TriSurface.load(meshfile,
inflatedmeshpath=inflatedmeshpath,
annotfile=annotfile)
self.assertTrue(numpy.allclose(self.kwargs["vertices"], surf.vertices))
self.assertTrue(
numpy.allclose(self.kwargs["triangles"], surf.triangles))
self.assertTrue(
numpy.allclose(self.kwargs["inflated_vertices"],
surf.inflated_vertices))
self.assertEqual(surf.shape(), (6, 18, 8))