update; fix tests

Exabyte-io · Oct 3, 2024 · 15c50a0 · 15c50a0
1 parent d588527
commit 15c50a0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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())