def test_vtk_set_dataset_wrong_datatype(self): mesh_points = fh.FileHandler( 'tests/test_datasets/matlab_field_test_bin.vtk').get_dataset( 'Pressure') with self.assertRaises(ValueError): fh.FileHandler( 'tests/test_datasets/matlab_field_test_bin.vtk').set_dataset( mesh_points, 'Pressure', 'faces')
def test_vtk_write0(self): outfilename = 'tests/test_datasets/matlab_field_test_bin_write.vtk' read_handl = fh.FileHandler( 'tests/test_datasets/matlab_field_test_bin.vtk') points, cells = read_handl.get_geometry(get_cells=True) write_handl = fh.FileHandler(outfilename) write_handl.set_geometry(points, cells) os.remove(outfilename)
def test_stl_write_comparison(self): output = fh.FileHandler(mat_file).get_dataset('output') output[0] = [1.1] output[1] = [1.1] output[2] = [1.1] output[11] = [1.1] output[12] = [1.1] output[13] = [1.1] output[30] = [1.1] output[31] = [1.1] output[32] = [1.1] outfilename = 'tests/test_datasets/matlab_output_test_out.mat' fh.FileHandler(outfilename).set_dataset(output, 'output') os.remove(outfilename)
def test_vtk_list_output_point(self): read_handl = fh.FileHandler( 'tests/test_datasets/matlab_field_test_bin.vtk' ) out_point, out_cell = read_handl.get_all_output_names() expected_out = ['Velocity', 'Pressure'] assert (out_point == expected_out)
def test_vtk_list_output_cell(self): read_handl = fh.FileHandler( 'tests/test_datasets/openfoam_output_test.vtk' ) out_point, out_cell = read_handl.get_all_output_names() expected_out = ['cellID', 'p', 'U'] assert (out_cell == expected_out)
def test_vtk_write_failing_filename_type(self): handler = fh.FileHandler( 'tests/test_datasets/matlab_field_test_bin.vtk' ) output = handler.get_dataset('Pressure') with self.assertRaises(TypeError): handler.set_dataset(output, 4.)
def test_stl_set_geometry_comparison(self): mesh_points, cells = fh.FileHandler(stl_file).get_geometry( get_cells=True) mesh_points[0] = [-40.2, -20.5, 60.9] mesh_points[1] = [-40.2, -10.5, 60.9] mesh_points[2] = [-40.2, -10.5, 60.9] mesh_points[50] = [-40.2, -20.5, 60.9] mesh_points[51] = [-40.2, -10.5, 60.9] mesh_points[52] = [-40.2, -10.5, 60.9] mesh_points[126] = [-40.2, -20.5, 60.9] mesh_points[127] = [-40.2, -10.5, 60.9] mesh_points[128] = [-40.2, -10.5, 60.9] outfilename = 'tests/test_datasets/test_sphere_out.stl' outfilename_expected = 'tests/test_datasets/test_sphere_out_true.stl' fh.FileHandler(outfilename).set_geometry(mesh_points, cells) self.assertTrue(filecmp.cmp(outfilename, outfilename_expected)) os.remove(outfilename)
def test_stl_get_geometry_coords_5_bin(self): mesh_points = fh.FileHandler(stl_file).get_geometry()[0] np.testing.assert_almost_equal(mesh_points[-1, 2], -0.45048400)
def test_mat_parse_failing_output_name_type(self): with self.assertRaises(TypeError): output = fh.FileHandler(mat_file).get_dataset(5.2)
def test_stl_get_geometry_coords_2(self): mesh_points = fh.FileHandler(stl_file).get_geometry()[0] expected_coordinates = np.array([-0.25, -0.433012992, 0]) np.testing.assert_array_almost_equal(mesh_points[147], expected_coordinates)
def test_stl_get_geometry_coords_3(self): cells = fh.FileHandler(stl_file).get_geometry(get_cells=True)[1] expected_list = [19, 17, 20] assert (expected_list == cells[17])
def test_file_available_ext(self): with self.assertRaises(TypeError): handler = fh.FileHandler('inexistent_file.pdf')
def test_base_class_infile(self): file_handler = fh.FileHandler() assert file_handler.infile == None
def test_mat_get_dataset_coords_1(self): output = fh.FileHandler(mat_file).get_dataset('output') np.testing.assert_almost_equal(output[33][0], 16.8143769)
def test_vtk_instantiation2(self): vtk_handler = fh.FileHandler(vtk_file) assert (isinstance(vtk_handler, vh.VtkHandler))
def test_vtk_parse_failing_output_name_type(self): with self.assertRaises(TypeError): output = fh.FileHandler( 'tests/test_datasets/openfoam_output_test.vtk').get_dataset(5.2)
def test_mat_instantiation1(self): mat_handler = fh.FileHandler(mat_file) assert (isinstance(mat_handler, mh.MatlabHandler))
def test_vtk_parse_coords_2(self): output = fh.FileHandler('tests/test_datasets/matlab_field_test_bin.vtk' ).get_dataset('Pressure') np.testing.assert_almost_equal(output[0], 8.2308226)
def test_vtk_parse_check_data_format_2(self): output = fh.FileHandler( 'tests/test_datasets/openfoam_output_test.vtk').get_dataset( 'p', 'cell') assert output.shape == (400, 1)
def test_vtk_parse_shape(self): output = fh.FileHandler('tests/test_datasets/matlab_field_test_bin.vtk' ).get_dataset('Pressure') assert output.shape == (2500, 1)
def test_mat_parse_failing_filename_type(self): with self.assertRaises(TypeError): fh.FileHandler(5.2)
def test_stl_get_geometry_wrong_filename(self): with self.assertRaises(RuntimeError): mesh_points = fh.FileHandler('nonexistent.stl').get_geometry()
def test_mat_get_dataset_shape(self): output = fh.FileHandler(mat_file).get_dataset('output') assert output.shape == (64, 1)
def test_stl_instantiation(self): stl_handler = fh.FileHandler(stl_file) assert (isinstance(stl_handler, sh.StlHandler))
def test_mat_get_dataset_coords_2(self): output = fh.FileHandler(mat_file).get_dataset('output') np.testing.assert_almost_equal(output[0][0], 149.0353302)
def test_stl_shape_point(self): mesh_points = fh.FileHandler(stl_file).get_geometry()[0] assert mesh_points.shape == (197, 3)
def test_file_wrong_type_filename(self): with self.assertRaises(TypeError): handler = fh.FileHandler([3])
def test_stl_shape_cell(self): cells = fh.FileHandler(stl_file).get_geometry(get_cells=True)[1] assert len(cells) == 390
def test_file_filename(self): handler = fh.FileHandler('inexistent_file.vtk') assert (handler._filename == 'inexistent_file.vtk')
def test_stl_coords_1(self): mesh_points = fh.FileHandler(stl_file).get_geometry()[0] np.testing.assert_almost_equal(mesh_points[33, 0], 0.198186)