def test_write_parameters_filename_default(self): params = IDWParameters() outfilename = 'parameters_rbf.prm' outfilename_expected = 'tests/test_datasets/parameters_idw_default.prm' params.write_parameters(outfilename) self.assertTrue(filecmp.cmp(outfilename, outfilename_expected)) os.remove(outfilename)
def test_idw_perform_deform(self): params = IDWParameters() expected_stretch = [1.19541593, 1.36081491, 1.42095073] params.read_parameters('tests/test_datasets/parameters_idw_deform.prm') idw = IDW(params, self.get_cube_mesh_points()) idw.perform() np.testing.assert_array_almost_equal(idw.modified_mesh_points[-3], expected_stretch)
def test_read_deformed(self): params = IDWParameters() filename = 'tests/test_datasets/parameters_idw_deform.prm' def_vertices = np.array([[0., 0., 0.], [0., 0., 1.], [0., 1., 0.], [1., 0., 0.], [0., 1., 1.], [1., 0., 1.], [1., 1., 0.], [1.5, 1.6, 1.7]]) params.read_parameters(filename) np.testing.assert_equal(params.deformed_control_points, def_vertices)
def test_idw_perform(self): params = IDWParameters() params.read_parameters('tests/test_datasets/parameters_idw_default.prm') IDW(params, self.get_cube_mesh_points()).perform()
def test_read_not_real_file(self): params = IDWParameters() with self.assertRaises(IOError): params.read_parameters('not_real_file')
def test_read_not_string(self): params = IDWParameters() with self.assertRaises(TypeError): params.read_parameters(5)
def test_read_p(self): params = IDWParameters() filename = 'tests/test_datasets/parameters_idw_deform.prm' params.read_parameters(filename) assert params.power == 3
def test_class_members_default_deformed_points(self): params = IDWParameters() cube_vertices = 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.]]) np.testing.assert_equal(params.deformed_control_points, cube_vertices)