Add missing imports to __init__

Author: FranzBangar

src/classy_blocks/__init__.py

@@ -1,11 +1,13 @@
 from .base.transforms import Mirror, Rotation, Scaling, Shear, Translation
+from .construct.assemblies.assembly import Assembly
 from .construct.assemblies.joints import LJoint, NJoint, TJoint
 from .construct.curves.analytic import AnalyticCurve, CircleCurve, LineCurve
 from .construct.curves.curve import CurveBase
 from .construct.curves.discrete import DiscreteCurve
 from .construct.curves.interpolated import LinearInterpolatedCurve, SplineInterpolatedCurve
 from .construct.edges import Angle, Arc, OnCurve, Origin, PolyLine, Project, Spline
 from .construct.flat.face import Face
+from .construct.flat.sketch import Sketch
 from .construct.flat.sketches.disk import FourCoreDisk, HalfDisk, OneCoreDisk, Oval, WrappedDisk
 from .construct.flat.sketches.grid import Grid
 from .construct.flat.sketches.mapped import MappedSketch
@@ -51,6 +53,7 @@
     "AnalyticCurve",
     "Angle",
     "Arc",
+    "Assembly",
     "Box",
     "CircleCurve",
     "ClampBase",
@@ -117,6 +120,7 @@
     "Shear",
     "Shell",
     "SimpleGrader",
+    "Sketch",
     "SketchOptimizer",
     "SketchSmoother",
     "SmoothGrader",

tests/test_util/test_imports.py

@@ -92,11 +92,13 @@ def test_import_links(self):
 
     def test_import_optimizer(self):
         _ = cb.MeshOptimizer
+        _ = cb.ShapeOptimizer
         _ = cb.SketchOptimizer
         _ = cb.MeshSmoother
         _ = cb.SketchSmoother
 
     def test_import_assemblies(self):
+        _ = cb.Assembly
         _ = cb.NJoint
         _ = cb.TJoint
         _ = cb.LJoint