4199 sweep along vector issue (#4250)

* Fix issue in sweep_along_vector() when managing list of object as input. Update UT test_47_sweep_along_vector() additional lines * Fix issue in sweep_along_vector() when managing list of object as input. Update UT test_47_sweep_along_vector() additional lines * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * revert unexpected change in sweep_around_axis() * Fix docstring syntax * Update pyaedt/modeler/cad/Primitives.py Co-authored-by: Kathy Pippert <[email protected]> --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Kathy Pippert <[email protected]>
ansys · Feb 21, 2024 · a2b448d · a2b448d
1 parent 500ef87
commit a2b448d
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/_unittest/example_models/cad/GDS/ICpackage_IC_on_board_Done_IC_Spirale.aedbcomp b/_unittest/example_models/cad/GDS/ICpackage_IC_on_board_Done_IC_Spirale.aedbcomp
diff --git a/_unittest/test_02_3D_modeler.py b/_unittest/test_02_3D_modeler.py
@@ -669,6 +669,14 @@ def test_47_sweep_along_vector(self):
         sweep_vector = [5, 0, 0]
         rect1 = self.aedtapp.modeler.create_rectangle(self.aedtapp.PLANE.YZ, [0, 0, 0], [20, 20], "rectangle_to_vector")
         assert rect1.sweep_along_vector(sweep_vector)
+        rect2 = self.aedtapp.modeler.create_rectangle(
+            self.aedtapp.PLANE.YZ, [0, 40, 0], [20, 20], "rectangle_to_vector2"
+        )
+        rect3 = self.aedtapp.modeler.create_rectangle(
+            self.aedtapp.PLANE.YZ, [0, 80, 0], [20, 20], "rectangle_to_vector3"
+        )
+        rect_list = [rect2, rect3]
+        assert self.aedtapp.modeler.sweep_along_vector(objid=rect_list, sweep_vector=sweep_vector)
 
     def test_48_coordinate_systems_parametric(self):
         self.aedtapp["var1"] = "5mm"

diff --git a/pyaedt/modeler/cad/Primitives.py b/pyaedt/modeler/cad/Primitives.py
@@ -2863,8 +2863,8 @@ def sweep_along_vector(self, objid, sweep_vector, draft_angle=0, draft_type="Rou
 
         Returns
         -------
-        bool
-            ``True`` when successful, ``False`` when failed.
+        :class:`pyaedt.modeler.cad.object3d.Object3d` or list of :class:`pyaedt.modeler.cad.object3d.Object3d`
+            One or more objects created.
 
         References
         ----------
@@ -2883,7 +2883,13 @@ def sweep_along_vector(self, objid, sweep_vector, draft_angle=0, draft_type="Rou
 
         self.oeditor.SweepAlongVector(vArg1, vArg2)
 
-        return self.update_object(objid)
+        if isinstance(objid, list):
+            updated_obj = []
+            for sel_obj in objid:
+                updated_obj.append(self.update_object(sel_obj))
+            return updated_obj
+        else:
+            return self.update_object(objid)
 
     @pyaedt_function_handler()
     def sweep_along_path(