From 15c50a0ff5c6b910e60cd4ddfeb403a706e31475 Mon Sep 17 00:00:00 2001 From: VsevolodX <79542055+VsevolodX@users.noreply.github.com> Date: Thu, 3 Oct 2024 13:29:50 -0700 Subject: [PATCH] update; fix tests --- tests/py/unit/test_tools_modify.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/py/unit/test_tools_modify.py b/tests/py/unit/test_tools_modify.py index bf53e2d3..8446a594 100644 --- a/tests/py/unit/test_tools_modify.py +++ b/tests/py/unit/test_tools_modify.py @@ -5,13 +5,13 @@ from mat3ra.made.tools.convert.utils import InterfacePartsEnum from mat3ra.made.tools.modify import ( add_vacuum, - displace_interface_part, filter_by_circle_projection, filter_by_label, filter_by_layers, filter_by_rectangle_projection, filter_by_sphere, filter_by_triangle_projection, + interface_displace_part, remove_vacuum, rotate, translate_to_z_level, @@ -185,7 +185,7 @@ def test_displace_interface(): {"id": 4, "value": [0.766666667, 0.866666667, 0.911447347]}, ] expected_labels = GRAPHENE_NICKEL_INTERFACE["basis"]["labels"] - displaced_material = displace_interface_part( + displaced_material = interface_displace_part( material, [0.1, 0.2, 0.3], InterfacePartsEnum.FILM, use_cartesian_coordinates=False ) assertion_utils.assert_deep_almost_equal(expected_coordinates, displaced_material.basis.coordinates.to_dict()) @@ -206,6 +206,6 @@ def test_displace_interface_optimized(): optimal_displacement = get_optimal_film_displacement( material, grid_size_xy=(10, 10), grid_range_x=(-0.5, 0.5), grid_range_y=(-0.5, 0.5) ) - displaced_material = displace_interface_part(material, optimal_displacement, use_cartesian_coordinates=True) + displaced_material = interface_displace_part(material, optimal_displacement, use_cartesian_coordinates=True) assertion_utils.assert_deep_almost_equal(expected_coordinates, displaced_material.basis.coordinates.to_dict()) assertion_utils.assert_deep_almost_equal(expected_labels, displaced_material.basis.labels.to_dict())