def test_mapper_find_neighbour_coord(self): mapper = ma.Mapper() points = np.array([[0, 0, 0], [0, 0, 1], [0, 0, 2]]) mapper._build_neighbour_locator(points) neighbour = mapper._find_neighbour([1, 0, 0]) closest = points[neighbour[:, 0].astype(int)] np.testing.assert_array_almost_equal(closest, [points[0]])
def test_mapper_find_neighbour_dist(self): mapper = ma.Mapper() points = np.array([[0, 0, 0], [0, 0, 1], [0, 0, 2]]) mapper._build_neighbour_locator(points) neighbour = mapper._find_neighbour([1, 0, 0]) closest = neighbour[:, mapper.number_neighbors:] np.testing.assert_almost_equal(1., closest)
def test_mapper_set_output_append2(self): mapper = ma.Mapper() names = ["this", "is"] names2 = ["a", "test"] mapper.output_name = names mapper.output_name = mapper.output_name + names2 assert ((names + names2) == mapper.output_name)
def test_mapper_map_solution(self): mapper = ma.Mapper() mapper.output_name = "test" mapper.map_solution(map_file, vtk_file, stl_file) test_map = FileHandler(map_file).get_dataset("test") os.remove(map_file) assert (test_map.shape == (197, 1))
def test_mapper_set_output_append4(self): mapper = ma.Mapper() names = "this" names2 = "a" mapper.output_name = names mapper.output_name = mapper.output_name + [names2] assert (["this", "a"] == mapper.output_name)
def test_mapper_map_solution3(self): mapper = ma.Mapper() mapper.output_name = "test" mapper.map_solution(map_file, vtk_file, stl_file) test_map = FileHandler(map_file).get_dataset("test") os.remove(map_file) np.testing.assert_almost_equal(test_map[117, 0], 0.71308929, decimal=3)
def test_mapper_map_solution_name(self): mapper = ma.Mapper() mapper.output_name = "test" mapper.map_solution(map_file, vtk_file, stl_file) point_output_name = FileHandler(map_file).get_all_output_names()[0] os.remove(map_file) assert (point_output_name == ["test"])
def test_mapper_map_solution2(self): mapper = ma.Mapper() mapper.output_name = "test" mapper.interpolation_mode = 'cell' mapper.map_solution(map_file, vtk_file, stl_file) test_map = FileHandler(map_file).get_dataset("test", datatype='cell') os.remove(map_file) assert (test_map.shape == (390, 1))
def test_mapper_map_solution_cell(self): mapper = ma.Mapper() mapper.output_name = "test" mapper.map_solution(map_file, vtk_file, stl_file) cell_map = FileHandler(map_file).get_geometry(get_cells=True)[1] cell_stl = FileHandler(stl_file).get_geometry(get_cells=True)[1] os.remove(map_file) assert (cell_map == cell_stl)
def test_mapper_map_solution4(self): mapper = ma.Mapper() mapper.output_name = "test" mapper.interpolation_mode = 'cell' mapper.map_solution(map_file, vtk_file, stl_file) test_map = FileHandler(map_file).get_dataset("test", datatype='cell') np.testing.assert_almost_equal(test_map[253, 0], 766.0) os.remove(map_file)
def test_mapper_map_solution_point(self): mapper = ma.Mapper() mapper.output_name = "test" mapper.map_solution(map_file, vtk_file, stl_file) point_map = FileHandler(map_file).get_geometry()[0] point_stl = FileHandler(stl_file).get_geometry()[0] os.remove(map_file) np.testing.assert_array_almost_equal(point_map, point_stl)
def test_find_neighbour(self): mapper = ma.Mapper() with self.assertRaises(RuntimeError): mapper._find_neighbour([0, 0, 0])
def test_mapper_number_neighbors_setter_wrongtype(self): mapper = ma.Mapper() with self.assertRaises(TypeError): mapper.number_neighbors = np.array([3])
def test_mapper_number_neighbors(self): mapper = ma.Mapper() mapper.number_neighbors = 3 assert (3 == mapper.number_neighbors)
def test_interpolation_mode_setter_wrongvalue(self): mapper = ma.Mapper() with self.assertRaises(ValueError): mapper.interpolation_mode = 'node'
def test_interpolation_mode_setter_wrongtype(self): mapper = ma.Mapper() with self.assertRaises(TypeError): mapper.interpolation_mode = 5
def test_interpolate_function_setter(self): def custom_interp(values, dist): return values[np.argmin(dist)] mapper = ma.Mapper() mapper.interpolate_function = custom_interp
def test_mapper_instantiation(self): mapper = ma.Mapper()
def test_mapper_map_solution_nogeometryfile(self): mapper = ma.Mapper() mapper.output_name = "Pressure" mapper.interpolation_mode = 'point' mapper.map_solution(vtk_file, 'tests/test_datasets/matlab_00.vtk')
def test_mapper_callable_build_neighbour_locator(self): mapper = ma.Mapper() points = np.eye(3) mapper._build_neighbour_locator(points)
def test_mapper_set_output(self): mapper = ma.Mapper() names = ["this", "is", "a", "test"] mapper.output_name = names assert (names == mapper.output_name)
def test_mapper_set_output_wrong_type(self): mapper = ma.Mapper() names = ["this", "is", 5.2, "test"] with self.assertRaises(TypeError): mapper.output_name = names
def test_mapper_map_solution_wrong_name(self): mapper = ma.Mapper() mapper.output_name = "testtest" with self.assertRaises(RuntimeError): mapper.map_solution(map_file, vtk_file, stl_file)