def test_rbf_weights_member(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_cube.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
weights_true = np.load('tests/test_datasets/weights_rbf_cube.npy')
np.testing.assert_array_almost_equal(rbf.weights, weights_true)
def test_rbf_original_mesh_points_member(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
np.testing.assert_array_almost_equal(rbf.original_mesh_points,
self.get_cube_mesh_points())
def test_utils_plot_rbf_control_points_save_fig(self):
params = pars.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_cube.prm')
ut.plot_rbf_control_points(params, save_fig=True)
self.assertTrue(os.path.isfile('RBF_control_points.png'))
os.remove('RBF_control_points.png')
def test_beckert_wendland_c2_basis_02(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.beckert_wendland_c2_basis(
np.array([0.1, 0.15, -0.2]).reshape(3, 1), 0.9)
np.testing.assert_almost_equal(value, 0.529916819595)
def test_read_parameters_filename_default(self):
params = rbfp.RBFParameters()
params.read_parameters()
outfilename = 'parameters_rbf.prm'
outfilename_expected = 'tests/test_datasets/parameters_rbf_default.prm'
self.assertTrue(filecmp.cmp(outfilename, outfilename_expected))
os.remove(outfilename)
def test_multi_quadratic_biharmonic_spline(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.multi_quadratic_biharmonic_spline(
np.linalg.norm(np.array([0.5, 1, 2, 0.2])), 0.2)
np.testing.assert_almost_equal(value, 2.30867927612)
def test_gaussian_spline(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.gaussian_spline(
np.array([0.5, 1, 2, 0.2]).reshape(4, 1), 0.2)
np.testing.assert_almost_equal(value, 0.0)
def test_beckert_wendland_c2_basis_01(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.beckert_wendland_c2_basis(
np.linalg.norm(np.array([0.5, 1, 2, 0.2])), 0.2)
np.testing.assert_almost_equal(value, 0.0)
def test_thin_plate_spline(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.thin_plate_spline(
np.linalg.norm(np.array([0.5, 1, 2, 0.2])), 0.2)
np.testing.assert_almost_equal(value, 323.000395428)
def test_read_parameters_deformed_control_points(self):
params = rbfp.RBFParameters()
params.read_parameters(
'tests/test_datasets/parameters_rbf_default.prm')
deformed_control_points_exact = np.array([0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., \
0., 1., 1., 1., 0., 1., 1., 1., 0., 1., 1., 1.]).reshape((8, 3))
np.testing.assert_array_almost_equal(params.deformed_control_points,
deformed_control_points_exact)
def test_inv_multi_quadratic_biharmonic_spline(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.inv_multi_quadratic_biharmonic_spline(
np.array([0.5, 1, 2, 0.2]).reshape(4, 1), 0.2)
np.testing.assert_almost_equal(value, 0.433148081824)
def test_polyharmonic_spline_k_even(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
params.power = 3
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.polyharmonic_spline(
np.array([0.1, 0.15, -0.2]).reshape(3, 1), 0.9)
np.testing.assert_almost_equal(value, 0.02677808)
def test_polyharmonic_spline_k_odd2(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
params.power = 2
rbf = rad.RBF(params, self.get_cube_mesh_points())
value = rbf.polyharmonic_spline(
np.linalg.norm(np.array([0.1, 0.15, -0.2])), 0.2)
np.testing.assert_almost_equal(value, 0.53895331)
def test_rbf_cube_mod(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_cube.prm')
mesh_points_ref = np.load(
'tests/test_datasets/meshpoints_cube_mod_rbf.npy')
rbf = rad.RBF(params, self.get_cube_mesh_points())
rbf.perform()
mesh_points_test = rbf.modified_mesh_points
np.testing.assert_array_almost_equal(mesh_points_test, mesh_points_ref)
def test_write_parameters(self):
params = rbfp.RBFParameters()
params.read_parameters('tests/test_datasets/parameters_rbf_cube.prm')
outfilename = 'tests/test_datasets/parameters_rbf_cube_out.prm'
outfilename_expected = 'tests/test_datasets/parameters_rbf_cube_out_true.prm'
params.write_parameters(outfilename)
self.assertTrue(filecmp.cmp(outfilename, outfilename_expected))
os.remove(outfilename)
def test_write_parameters_filename_default_existance(self):
params = rbfp.RBFParameters()
params.basis = 'inv_multi_quadratic_biharmonic_spline'
params.radius = 0.1
params.n_control_points = 3
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)
def test_write_parameters_filename_default(self):
params = rbfp.RBFParameters()
params.basis = 'gaussian_spline'
params.radius = 0.5
params.n_control_points = 8
params.original_control_points = np.array([0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., \
0., 1., 1., 1., 0., 1., 1., 1., 0., 1., 1., 1.]).reshape((8, 3))
params.deformed_control_points = np.array([0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., \
0., 1., 1., 1., 0., 1., 1., 1., 0., 1., 1., 1.]).reshape((8, 3))
params.write_parameters()
outfilename = 'parameters_rbf.prm'
outfilename_expected = 'tests/test_datasets/parameters_rbf_default.prm'
self.assertTrue(filecmp.cmp(outfilename, outfilename_expected))
os.remove(outfilename)
def test_utils_plot_rbf_control_points(self):
params = pars.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_cube.prm')
ut.plot_rbf_control_points(params, save_fig=False)
def test_read_parameters_radius(self):
params = rbfp.RBFParameters()
params.read_parameters(
'tests/test_datasets/parameters_rbf_default.prm')
assert params.radius == 0.5
def test_class_members_default_radius(self):
params = rbfp.RBFParameters()
assert params.radius == None
def test_wrong_basis(self):
params = rbfp.RBFParameters()
params.read_parameters(
'tests/test_datasets/parameters_rbf_bugged_02.prm')
with self.assertRaises(NameError):
rbf = rad.RBF(params, self.get_cube_mesh_points())
def test_rbf_default_modified_mesh_points_member(self):
params = rbfp.RBFParameters()
params.read_parameters(
filename='tests/test_datasets/parameters_rbf_default.prm')
rbf = rad.RBF(params, self.get_cube_mesh_points())
assert rbf.modified_mesh_points == None
def test_print_info(self):
params = rbfp.RBFParameters()
print(params)
def test_class_members_default_deformed_control_points(self):
params = rbfp.RBFParameters()
assert params.deformed_control_points == None
def test_class_members_default_original_control_points(self):
params = rbfp.RBFParameters()
assert params.original_control_points == None
def test_class_members_default_basis(self):
params = rbfp.RBFParameters()
assert params.basis == None
def test_read_parameters_basis(self):
params = rbfp.RBFParameters()
params.read_parameters(
'tests/test_datasets/parameters_rbf_default.prm')
assert params.basis == 'gaussian_spline'
def test_write_parameters_failing_filename_type(self):
params = rbfp.RBFParameters()
with self.assertRaises(TypeError):
params.write_parameters(5)
def test_read_parameters_failing_number_deformed_control_points(self):
params = rbfp.RBFParameters()
with self.assertRaises(TypeError):
params.read_parameters(
'tests/test_datasets/parameters_rbf_bugged_01.prm')
def test_read_parameters_n_control_points(self):
params = rbfp.RBFParameters()
params.read_parameters(
'tests/test_datasets/parameters_rbf_default.prm')
assert params.n_control_points == 8
