Update type hinting and task defaults (#52)

* Update default for extract mesh projections task

* Remove None type hinting in tests

src/abm_shape_collection/extract_mesh_projections.py

@@ -27,7 +27,7 @@ class ProjectionType(Enum):
 
 def extract_mesh_projections(
     mesh: vtkPolyData | trimesh.Trimesh,
-    projection_types: list[ProjectionType],
+    projection_types: list[ProjectionType] | None = None,
     offset: tuple[float, float, float] | None = None,
 ) -> dict:
     """
@@ -56,6 +56,9 @@ def extract_mesh_projections(
     if isinstance(mesh, vtkPolyData):
         mesh = convert_vtk_to_trimesh(mesh)
 
+    if projection_types is None:
+        projection_types = [ProjectionType.SLICE, ProjectionType.EXTENT]
+
     if offset is not None:
         mesh.apply_translation(offset)
 

tests/abm_shape_collection/test_extract_mesh_projections.py

@@ -14,7 +14,7 @@
 
 
 class TestExtractMeshProjections(unittest.TestCase):
-    def setUp(self) -> None:
+    def setUp(self):
         vertices = [
             (-1, -1, 0),
             (1, -1, 0),
@@ -116,6 +116,14 @@ def setUp(self) -> None:
             },
         }
 
+    def test_extract_mesh_projections_all_types_no_translation(self):
+        projections = extract_mesh_projections(self.vtk_mesh)
+        for proj, _, _ in PROJECTIONS:
+            self.assertCountEqual(self.slices[proj], projections[f"{proj}_slice"])
+            self.assertDictEqual(self.extents[proj], projections[f"{proj}_extent"])
+            self.assertFalse(f"{proj}_extent" in proj)
+            self.assertFalse(f"{proj}_slice" in proj)
+
     def test_extract_mesh_projections_slices_no_translation(self):
         projections = extract_mesh_projections(self.vtk_mesh, [ProjectionType.SLICE], offset=None)
         for proj, _, _ in PROJECTIONS:

tests/abm_shape_collection/test_extract_mesh_wireframe.py

@@ -7,7 +7,7 @@
 
 
 class TestExtractMeshProjections(unittest.TestCase):
-    def setUp(self) -> None:
+    def setUp(self):
         vertices = [
             (-1, -1, 0),
             (1, -1, 0),

tests/abm_shape_collection/test_get_shape_coefficients.py

@@ -6,7 +6,7 @@
 
 
 class TestGetShapeCoefficients(unittest.TestCase):
-    def setUp(self) -> None:
+    def setUp(self):
         self.order = 2
 
         self.coeff_names = [