def test_extract_image_plane(sorted_nodes, axes):
from fem.post.create_res_sim import extract_image_plane
image_plane = extract_image_plane(sorted_nodes, axes, 0.0)
assert image_plane.shape == (11, 11)
assert image_plane[0][0] == 11
assert image_plane[0][-1] == 1221
assert image_plane[-1][0] == 121
assert image_plane[-1][-1] == 1331
# the ele_post kwarg is deprecated and should now raise a TypeError if
# passed
with pytest.raises(TypeError):
image_plane = extract_image_plane(sorted_nodes, axes, ele_pos=0.0)
assert image_plane.shape == (11, 11)
assert image_plane[0][0] == 11
assert image_plane[0][-1] == 1221
assert image_plane[-1][0] == 121
assert image_plane[-1][-1] == 1331
image_plane = extract_image_plane(sorted_nodes,
axes,
plane_pos=0.0,
direction=1)
assert image_plane.shape == (11, 11)
assert image_plane[0][0] == 1
assert image_plane[0][-1] == 1211
assert image_plane[-1][0] == 11
assert image_plane[-1][-1] == 1221
image_plane = extract_image_plane(sorted_nodes,
axes,
plane_pos=0.0,
direction=2)
assert image_plane.shape == (11, 11)
assert image_plane[0][0] == 1211 #11
assert image_plane[0][-1] == 1321 #121
assert image_plane[-1][0] == 1221 #1
assert image_plane[-1][-1] == 1331 #111
with pytest.raises(ValueError):
image_plane = extract_image_plane(sorted_nodes,
axes,
plane_pos=0.0,
direction=3)
def test_extract_arfi_data():
"""test extraction of arfi data at specific timesteps
"""
from fem.post.create_res_sim import extract_arfi_data
from fem.post.create_res_sim import read_header
from fem.post.create_res_sim import extract_image_plane
from fem.mesh.fem_mesh import load_nodeIDs_coords
from fem.mesh.fem_mesh import SortNodeIDs
from scipy.io import loadmat
import numpy as np
gauss_qsym_pml_example_path = examplesPath / "gauss_qsym_pml"
dispout = gauss_qsym_pml_example_path / "disp.dat.xz"
nodedyn = gauss_qsym_pml_example_path / "nodes.dyn"
valid_data = loadmat(gauss_qsym_pml_example_path / "res_sim_valid.mat")['arfidata']
header = read_header(dispout)
node_id_coords = load_nodeIDs_coords(nodedyn)
[snic, axes] = SortNodeIDs(node_id_coords)
image_plane = extract_image_plane(snic, axes)
test_data0 = extract_arfi_data(dispout, header, image_plane)
test_data1 = extract_arfi_data(dispout, header, image_plane, specific_times=[1, 3])
test_data2 = extract_arfi_data(dispout, header, image_plane, specific_times=[3, 2, 1])
assert np.array_equal(test_data0, valid_data)
assert np.array_equal(test_data1, valid_data[:, :, [0, 2]])
assert np.array_equal(test_data2, np.flip(valid_data, axis=-1))
def test_preallocate_arfidata(sorted_nodes, axes):
from fem.post.create_res_sim import __preallocate_arfidata
from fem.post.create_res_sim import extract_image_plane
image_plane = extract_image_plane(sorted_nodes, axes, 0.0)
arfidata = __preallocate_arfidata(image_plane, 3)
assert arfidata.shape == (11, 11, 3)
def test_extract_image_plane(sorted_nodes, axes):
from fem.post.create_res_sim import extract_image_plane
image_plane = extract_image_plane(sorted_nodes, axes, 0.0)
assert image_plane.shape == (11, 11)
assert image_plane[0][0] == 11
assert image_plane[0][-1] == 1221
assert image_plane[-1][0] == 121
assert image_plane[-1][-1] == 1331