Skip to content

Add tests for mesh data #249

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Aug 7, 2023
Merged

Add tests for mesh data #249

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e26b57a
Add tests for model mesh data
greschd Aug 3, 2023
ce54845
Add tests for pyvista conversion, modeling ply mesh data
greschd Aug 3, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
82 changes: 81 additions & 1 deletion tests/unittests/test_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,12 @@
import pathlib
import tempfile

from ansys.acp.core import Client
import numpy as np
import numpy.testing
import pytest
import pyvista

from ansys.acp.core import Client, ElementalDataType

from .helpers import check_property

Expand Down Expand Up @@ -120,3 +125,78 @@ def test_string_representation(grpc_server, model_data_dir):
model_str = str(model)
assert model_str.startswith("Model(")
assert f"name='{model.name}'" in model_str


@pytest.fixture
def minimal_complete_model(load_model_from_tempfile):
with load_model_from_tempfile() as model:
yield model


def test_mesh_data(minimal_complete_model):
mesh = minimal_complete_model.mesh

numpy.testing.assert_equal(mesh.node_labels, np.array([1, 2, 3, 4]))
numpy.testing.assert_allclose(
mesh.node_coordinates,
np.array(
[
[-150.0, 150.0, 0.0],
[-150.0, -150.0, 0.0],
[150.0, -150.0, 0.0],
[150.0, 150.0, 0.0],
]
),
)
numpy.testing.assert_equal(mesh.element_labels, np.array([1]))
numpy.testing.assert_equal(mesh.element_types, np.array([126]))
numpy.testing.assert_equal(mesh.element_nodes, np.array([0, 1, 2, 3]))
numpy.testing.assert_equal(mesh.element_nodes_offsets, np.array([0]))


def test_elemental_data(minimal_complete_model):
data = minimal_complete_model.elemental_data
numpy.testing.assert_allclose(data.element_labels, np.array([1]))
numpy.testing.assert_allclose(data.normal, np.array([[0.0, 0.0, 1.0]]))
numpy.testing.assert_allclose(data.thickness, np.array([1e-4]))
numpy.testing.assert_allclose(data.relative_thickness_correction, np.array([1.0]))
numpy.testing.assert_allclose(data.area, np.array([9e4]))
numpy.testing.assert_allclose(data.price, np.array([0.0]))
numpy.testing.assert_allclose(data.volume, np.array([9.0]))
numpy.testing.assert_allclose(data.mass, np.array([7.065e-08]))
numpy.testing.assert_allclose(data.offset, np.array([5e-5]))
numpy.testing.assert_allclose(data.cog, np.array([[0.0, 0.0, 5e-5]]))


def test_nodal_data(minimal_complete_model):
data = minimal_complete_model.nodal_data
numpy.testing.assert_allclose(data.node_labels, np.array([1, 2, 3, 4]))


def test_mesh_data_to_pyvista(minimal_complete_model):
pv_mesh = minimal_complete_model.mesh.to_pyvista()
assert isinstance(pv_mesh, pyvista.core.pointset.UnstructuredGrid)
assert pv_mesh.n_points == 4
assert pv_mesh.n_cells == 1


def test_elemental_data_to_pyvista(minimal_complete_model):
data = minimal_complete_model.elemental_data
pv_mesh = data.to_pyvista(mesh=minimal_complete_model.mesh)
assert isinstance(pv_mesh, pyvista.core.pointset.UnstructuredGrid)
assert pv_mesh.n_points == 4
assert pv_mesh.n_cells == 1


@pytest.mark.parametrize("component", [e.value for e in ElementalDataType])
def test_elemental_data_to_pyvista_with_component(minimal_complete_model, component):
data = minimal_complete_model.elemental_data
if not hasattr(data, component):
pytest.skip(f"Model elemental data does not contain component '{component}'")
pv_mesh = data.to_pyvista(mesh=minimal_complete_model.mesh, component=component)
if component in ["normal", "cog"]:
assert isinstance(pv_mesh, pyvista.core.pointset.PolyData)
else:
assert isinstance(pv_mesh, pyvista.core.pointset.UnstructuredGrid)
assert pv_mesh.n_points == 4
assert pv_mesh.n_cells == 1
142 changes: 142 additions & 0 deletions tests/unittests/test_modeling_ply.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@
import numpy as np
import numpy.testing
import pytest
import pyvista

from ansys.acp.core import ElementalDataType, NodalDataType

from .common.linked_object_list_tester import LinkedObjectListTestCase, LinkedObjectListTester
from .common.tree_object_tester import NoLockedMixin, ObjectPropertiesToTest, TreeObjectTester
Expand Down Expand Up @@ -79,3 +84,140 @@ def linked_object_case_nonempty(tree_object, parent_model):

class TestLinkedObjectLists(LinkedObjectListTester):
pass


@pytest.fixture
def minimal_complete_model(load_model_from_tempfile):
with load_model_from_tempfile() as model:
yield model


@pytest.fixture
def simple_modeling_ply(minimal_complete_model):
return minimal_complete_model.modeling_groups["ModelingGroup.1"].modeling_plies["ModelingPly.1"]


def test_elemental_data(simple_modeling_ply):
data = simple_modeling_ply.elemental_data
numpy.testing.assert_allclose(data.element_labels, np.array([1]))
numpy.testing.assert_allclose(data.normal, np.array([[0.0, 0.0, 1.0]]))

numpy.testing.assert_allclose(
data.orientation,
np.array([[0.0, 0.0, 1.0]]),
atol=1e-12,
)
numpy.testing.assert_allclose(
data.reference_direction,
np.array([[1.0, 0.0, 0.0]]),
atol=1e-12,
)
numpy.testing.assert_allclose(
data.fiber_direction,
np.array([[1.0, 0.0, 0.0]]),
atol=1e-12,
)
numpy.testing.assert_allclose(
data.draped_fiber_direction,
np.array([[1.0, 0.0, 0.0]]),
atol=1e-12,
)
numpy.testing.assert_allclose(
data.transverse_direction,
np.array([[0.0, 1.0, 0.0]]),
atol=1e-12,
)
numpy.testing.assert_allclose(
data.draped_transverse_direction,
np.array([[0.0, 1.0, 0.0]]),
atol=1e-12,
)

numpy.testing.assert_allclose(data.thickness, np.array([1e-4]))
numpy.testing.assert_allclose(data.relative_thickness_correction, np.array([1.0]))

numpy.testing.assert_allclose(data.design_angle, np.array([0.0]))
numpy.testing.assert_allclose(data.shear_angle, np.array([0.0]))
numpy.testing.assert_allclose(data.draped_fiber_angle, np.array([0.0]))
numpy.testing.assert_allclose(data.draped_transverse_angle, np.array([90.0]))

numpy.testing.assert_allclose(data.area, np.array([9e4]))
numpy.testing.assert_allclose(data.price, np.array([0.0]))
numpy.testing.assert_allclose(data.volume, np.array([9.0]))
numpy.testing.assert_allclose(data.mass, np.array([7.065e-08]))
numpy.testing.assert_allclose(data.offset, np.array([5e-5]))
numpy.testing.assert_allclose(data.cog, np.array([[0.0, 0.0, 5e-5]]))


def test_nodal_data(simple_modeling_ply):
data = simple_modeling_ply.nodal_data
numpy.testing.assert_allclose(data.node_labels, np.array([1, 2, 3, 4]))
numpy.testing.assert_allclose(
data.ply_offset,
np.array([[0.0, 0.0, 5e-5], [0.0, 0.0, 5e-5], [0.0, 0.0, 5e-5], [0.0, 0.0, 5e-5]]),
)


def test_elemental_data_to_pyvista(minimal_complete_model, simple_modeling_ply):
elemental_data = simple_modeling_ply.elemental_data
pv_mesh = elemental_data.to_pyvista(mesh=minimal_complete_model.mesh)
assert isinstance(pv_mesh, pyvista.core.pointset.UnstructuredGrid)
assert pv_mesh.n_points == 4
assert pv_mesh.n_cells == 1


@pytest.mark.parametrize("component", [e.value for e in ElementalDataType])
def test_elemental_data_to_pyvista_with_component(
minimal_complete_model, simple_modeling_ply, component
):
data = simple_modeling_ply.elemental_data
if not hasattr(data, component):
pytest.skip(f"Modeling Ply elemental data does not contain component '{component}'")
pv_mesh = data.to_pyvista(mesh=minimal_complete_model.mesh, component=component)
if component in [
"normal",
"orientation",
"reference_direction",
"fiber_direction",
"draped_fiber_direction",
"transverse_direction",
"draped_transverse_direction",
"cog",
]:
assert isinstance(
pv_mesh, pyvista.core.pointset.PolyData
), f"Created wrong mesh type PolyData for component '{component}'"
else:
assert isinstance(
pv_mesh, pyvista.core.pointset.UnstructuredGrid
), f"Created wrong mesh type UnstructuredGrid for component '{component}'"
assert pv_mesh.n_points == 4
assert pv_mesh.n_cells == 1


def test_nodal_data_to_pyvista(minimal_complete_model, simple_modeling_ply):
data = simple_modeling_ply.nodal_data
pv_mesh = data.to_pyvista(mesh=minimal_complete_model.mesh)
assert isinstance(pv_mesh, pyvista.core.pointset.UnstructuredGrid)
assert pv_mesh.n_points == 4
assert pv_mesh.n_cells == 1


@pytest.mark.parametrize("component", [e.value for e in NodalDataType])
def test_nodal_data_to_pyvista_with_component(
minimal_complete_model, simple_modeling_ply, component
):
data = simple_modeling_ply.nodal_data
if not hasattr(data, component):
pytest.skip(f"Modeling Ply nodal data does not contain component '{component}'")
pv_mesh = data.to_pyvista(mesh=minimal_complete_model.mesh, component=component)
if component in ["ply_offset"]:
assert isinstance(
pv_mesh, pyvista.core.pointset.PolyData
), f"Created wrong mesh type PolyData for component '{component}'"
else:
assert isinstance(
pv_mesh, pyvista.core.pointset.UnstructuredGrid
), f"Created wrong mesh type UnstructuredGrid for component '{component}'"
assert pv_mesh.n_points == 4
assert pv_mesh.n_cells == 1