Beispiel #1
0
 def test_write_parameters_filename_default_existance(self):
     params = RBFParameters()
     params.basis = 'inv_multi_quadratic_biharmonic_spline'
     params.radius = 0.1
     params.original_control_points = np.array(
         [0., 0., 0., 0., 0., 1., 0., 1., 0.]).reshape((3, 3))
     params.deformed_control_points = np.array(
         [0., 0., 0., 0., 0., 1., 0., 1., 0.]).reshape((3, 3))
     params.write_parameters()
     outfilename = 'parameters_rbf.prm'
     assert os.path.isfile(outfilename)
     os.remove(outfilename)
Beispiel #2
0
    def test_write_parameters_filename_default(self):
        params = RBFParameters()
        params.basis = 'gaussian_spline'
        params.radius = 0.5
        params.power = 2
        params.original_control_points = unit_cube
        params.deformed_control_points = unit_cube
        outfilename = 'test.prm'
        params.write_parameters(outfilename)
        outfilename_expected = 'tests/test_datasets/parameters_rbf_default.prm'

        print(filecmp.cmp(outfilename, outfilename_expected))
        self.assertTrue(filecmp.cmp(outfilename, outfilename_expected))
        os.remove(outfilename)