Restore transform star exports

Author: TonisOrmisson

src/openstatspec/transform/__init__.py

@@ -20,14 +20,15 @@
 __all__ = [
     "AssignOperation", "BooleanExpression", "ComparisonExpression",
     "ConditionalAssignOperation", "CreateVariableOperation", "DeleteVariableOperation",
-    "PredicateExpression",
+    "ExecuteOperation", "Operand", "PredicateExpression",
     "BoundTransformation",
     "RecodeMatch", "RecodeOperation", "RecodeResult",
     "RecodeRule", "ReplaceValueLabelsOperation", "SPSS_FRONTEND_CONTRACT",
     "SetFormatOperation", "SetMeasurementLevelOperation",
     "SetVariableLabelOperation", "SourcePosition", "SourceSpan", "StorageKind",
     "SpssFrontendCompilation",
-    "SpssSyntaxProgram", "TRANSFORMATION_PLAN_CONTRACT", "TransformationFrontendError",
+    "SpssSyntaxProgram", "TRANSFORMATION_PLAN_CONTRACT",
+    "TRANSFORMATION_PLAN_SCHEMA_CHANGE_CONTRACT", "TransformationFrontendError",
     "TransformationPlan", "TypedValue", "ValueLabel",
     "VariableDefinition",
     "VariableSchema", "bind_spss_syntax", "bind_transformation_plan",

tests/test_transform_frontend.py

@@ -5,6 +5,7 @@
 from pathlib import Path
 
 import pytest
+import openstatspec.transform as transform_module
 from openstatspec.frontends.spss import (
     bind_spss_syntax,
     compile_spss_syntax,
@@ -32,6 +33,14 @@
 )
 
 
+def test_transform_star_exports_keep_public_plan_nodes() -> None:
+    assert {
+        "ExecuteOperation",
+        "Operand",
+        "TRANSFORMATION_PLAN_SCHEMA_CHANGE_CONTRACT",
+    }.issubset(transform_module.__all__)
+
+
 def _schema(*variables: VariableDefinition) -> VariableSchema:
     return VariableSchema(tuple(variables))