Commit ffba26f6 authored by Benoit Martin's avatar Benoit Martin

fix unit tests

parent 0ff4e488
Pipeline #56327405 failed with stages
in 9 minutes and 30 seconds
...@@ -18,8 +18,8 @@ class DistanceTests(unittest.TestCase): ...@@ -18,8 +18,8 @@ class DistanceTests(unittest.TestCase):
def setUp(self): def setUp(self):
self.tensor_scalar_type = torch.DoubleTensor self.tensor_scalar_type = torch.DoubleTensor
self.tensor_integer_type = torch.LongTensor self.tensor_integer_type = torch.LongTensor
self.kernel = kernel_factory.factory('torch', 10.) self.kernel = kernel_factory.factory('torch', kernel_width=10.)
# self.kernel = kernel_factory.factory('keops', 10.) # Duplicate the tests for both kernels ? # self.kernel = kernel_factory.factory('keops', kernel_width=10.) # Duplicate the tests for both kernels ?
self.multi_attach = MultiObjectAttachment('', self.kernel) self.multi_attach = MultiObjectAttachment('', self.kernel)
def _read_surface_mesh(self, path): def _read_surface_mesh(self, path):
......
...@@ -37,7 +37,7 @@ class ParallelTransportTests(unittest.TestCase): ...@@ -37,7 +37,7 @@ class ParallelTransportTests(unittest.TestCase):
factor = 5 factor = 5
geodesic = Geodesic( geodesic = Geodesic(
dense_mode=False, dense_mode=False,
kernel=kernel_factory.factory('torch', 0.01), kernel=kernel_factory.factory('torch', kernel_width=0.01),
t0=0., t0=0.,
use_rk2_for_shoot=True, use_rk2_for_shoot=True,
concentration_of_time_points=10 * factor concentration_of_time_points=10 * factor
......
...@@ -25,7 +25,7 @@ class ShootingTests(unittest.TestCase): ...@@ -25,7 +25,7 @@ class ShootingTests(unittest.TestCase):
geodesic = Geodesic( geodesic = Geodesic(
dense_mode=False, dense_mode=False,
kernel=kernel_factory.factory('torch', 0.01), kernel=kernel_factory.factory('torch', kernel_width=0.01),
t0=0., t0=0.,
use_rk2_for_shoot=True, use_rk2_for_shoot=True,
use_rk2_for_flow=True, use_rk2_for_flow=True,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment