Move geometry_tree.py

Author: jgsdavies

src/ansys/motorcad/core/geometry.py

@@ -27,6 +27,7 @@
 from math import acos, atan2, cos, degrees, fabs, floor, inf, isclose, radians, sin, sqrt
 import warnings
 from warnings import warn
+
 from geometry_extrusion import ExtrusionBlockList
 
 GEOM_TOLERANCE = 1e-6
@@ -2421,7 +2422,6 @@ def _entities_same_with_direction(entities_1, entities_2):
             return _entities_same_with_direction(self, entities_to_compare)
 
 
-
 def _convert_entities_to_json(entities):
     """Get entities list as a json object.
 

src/ansys/motorcad/core/geometry_extrusion.py

@@ -1,7 +1,8 @@
+"""Geometry extrusion classes."""
+
 # Copyright (C) 2022 - 2025 ANSYS, Inc. and/or its affiliates.
 # SPDX-License-Identifier: MIT
 #
-#
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
 # in the Software without restriction, including without limitation the rights
@@ -20,6 +21,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
+
 class ExtrusionBlock:
     """Generic class for storing 3D extrusion data."""
 
@@ -105,7 +107,7 @@ def _to_json(self):
             "extrusion_block_start": self.start_pos,
             "extrusion_block_end": self.end_pos,
             "extrusion_block_angle_step": self.angle_step,
-            "extrusion_block_continuous_rotation": self.angle_continuous
+            "extrusion_block_continuous_rotation": self.angle_continuous,
         }
 
         return block_dict
@@ -158,4 +160,4 @@ def _from_json(self, json_list):
         for json_object in json_list:
             block = ExtrusionBlock()
             block._from_json(json_object)
-            self.append(block)
+            self.append(block)

tests/test_geometry.py

@@ -2619,7 +2619,7 @@ def test_region_find_entity_from_coordinates():
     assert c1.find_entity_from_coordinates(Coordinate(99, 99), Coordinate(99, 99)) is None
 
     assert (
-            c1.find_entity_from_coordinates(c1.entities[0].start, c1.entities[0].end) == c1.entities[0]
+        c1.find_entity_from_coordinates(c1.entities[0].start, c1.entities[0].end) == c1.entities[0]
     )
 
 
@@ -2819,19 +2819,19 @@ def test_set_lamination_type(mc_reset_to_default_on_teardown):
     assert rotor.lamination_type == "Solid"
 
     solid_rotor_section_file = (
-            get_dir_path() + r"\test_files\adaptive_template_testing_solid_rotor_region.mot"
+        get_dir_path() + r"\test_files\adaptive_template_testing_solid_rotor_region.mot"
     )
     lam_rotor_section_file = (
-            get_dir_path() + r"\test_files\adaptive_template_testing_lam_rotor_region.mot"
+        get_dir_path() + r"\test_files\adaptive_template_testing_lam_rotor_region.mot"
     )
 
     solid_rotor_section_result = (
-            get_dir_path() + r"\test_files\adaptive_template_testing_solid_rotor_region"
-                             r"\FEResultsData\StaticLoadInductance_result_1.mes"
+        get_dir_path() + r"\test_files\adaptive_template_testing_solid_rotor_region"
+        r"\FEResultsData\StaticLoadInductance_result_1.mes"
     )
     lam_rotor_section_result = (
-            get_dir_path() + r"\test_files\adaptive_template_testing_lam_rotor_region"
-                             r"\FEResultsData\StaticLoadInductance_result_1.mes"
+        get_dir_path() + r"\test_files\adaptive_template_testing_lam_rotor_region"
+        r"\FEResultsData\StaticLoadInductance_result_1.mes"
     )
 
     # load file into Motor-CAD
@@ -2855,6 +2855,3 @@ def test_region_creation_warnings(mc):
         _ = Region()
     with pytest.warns():
         _ = Region(mc)
-
-
-

tests/test_geometry_extrusion.py

@@ -28,7 +28,7 @@ def test_extrusion_block_to_json():
         "extrusion_block_start": 100.1,
         "extrusion_block_end": 200.2,
         "extrusion_block_angle_step": 1.5,
-        "extrusion_block_continuous_rotation": 0
+        "extrusion_block_continuous_rotation": 0,
     }
 
     block = ExtrusionBlock()
@@ -45,7 +45,7 @@ def test_extrusion_block_from_json():
         "extrusion_block_start": 100,
         "extrusion_block_end": 200,
         "extrusion_block_angle_step": 0,
-        "extrusion_block_continuous_rotation": 20
+        "extrusion_block_continuous_rotation": 20,
     }
     block = ExtrusionBlock()
     block._from_json(block_dict)
@@ -60,7 +60,7 @@ def test_extrusion_block_list_to_json():
             "extrusion_block_start": 100,
             "extrusion_block_end": 200,
             "extrusion_block_angle_step": 0,
-            "extrusion_block_continuous_rotation": 10
+            "extrusion_block_continuous_rotation": 10,
         }
     ]
 
@@ -77,7 +77,7 @@ def test_extrusion_block_list_from_json():
             "extrusion_block_start": 100,
             "extrusion_block_end": 200,
             "extrusion_block_angle_step": 0,
-            "extrusion_block_continuous_rotation": 20
+            "extrusion_block_continuous_rotation": 20,
         }
     ]
     blocks = ExtrusionBlockList()
@@ -90,4 +90,4 @@ def test_extrusion_block_list_from_json():
 
 def test_block_extrusion_length():
     block = ExtrusionBlock(start_pos=100, end_pos=200.5, angle_continuous=0)
-    assert block.extrusion_length == 100.5
+    assert block.extrusion_length == 100.5