py-teststand 0.2.2__tar.gz → 0.2.3__tar.gz

{py_teststand-0.2.2 → py_teststand-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-teststand
-Version: 0.2.2
+Version: 0.2.3
 Summary: Community object-oriented Python 3 bindings for the National Instruments TestStand™ COM API
 Project-URL: Repository, https://github.com/TheDomcio/py-teststand
 Project-URL: Issues, https://github.com/TheDomcio/py-teststand/issues

{py_teststand-0.2.2 → py_teststand-0.2.3}/examples/analyzer_step_name_length.py

@@ -37,7 +37,7 @@ MAXIMUM_STEP_NAME_LENGTH = 15
     transitions=(AnalysisTransition.BeforeStep,),
     severity=RuleSeverity.Warning,
 )
-def check_step_name_length(context: Any) -> None:
+def check_step_name_length(context: Any) -> None:  # noqa: ANN401
     """Report any step whose name is longer than MAXIMUM_STEP_NAME_LENGTH."""
     step: Any = context.object
     name: Any = step.name
@@ -51,7 +51,7 @@ def check_step_name_length(context: Any) -> None:
 
 
 class _Violation:
-    def __init__(self, rule_id: str, text: str, location: Any) -> None:
+    def __init__(self, rule_id: str, text: str, location: Any) -> None:  # noqa: ANN401
         self.rule_id: str = rule_id
         self.text: str = text
         self.location: Any = location
@@ -65,12 +65,12 @@ class _OfflineStepContext:
     py_teststand.analyzer.AnalysisContext instead, whose `object` is the step.
     """
 
-    def __init__(self, analyzed_object: Any) -> None:
+    def __init__(self, analyzed_object: Any) -> None:  # noqa: ANN401
         self.object: Any = analyzed_object
         self.transition = AnalysisTransition.BeforeStep
         self.findings: list[_Violation] = []
 
-    def report(self, rule_id: str, text: str, location_object: Any = None) -> None:
+    def report(self, rule_id: str, text: str, location_object: Any = None) -> None:  # noqa: ANN401
         self.findings.append(_Violation(rule_id, text, location_object))
 
 

{py_teststand-0.2.2 → py_teststand-0.2.3}/examples/property_object_serialize.py

@@ -26,7 +26,7 @@ INSERT_IF_MISSING = int(PropertyOption.InsertIfMissing)
 ROOT_TEMP_DIR = Path(tempfile.gettempdir()) / "py-teststand"
 
 
-def property_object_to_dict(container: Any) -> dict[str, Any]:
+def property_object_to_dict(container: Any) -> dict[str, Any]:  # noqa: ANN401
     """Walk a container PropertyObject's sub-properties into a plain dict."""
     result: dict[str, Any] = {}
     for index in range(container.get_num_sub_properties("")):
@@ -35,7 +35,7 @@ def property_object_to_dict(container: Any) -> dict[str, Any]:
     return result
 
 
-def _to_python(obj: Any) -> Any:
+def _to_python(obj: Any) -> Any:  # noqa: ANN401
     """Convert one sub-property object to a list (array), dict (container), or scalar."""
     value_type, _is_object, is_array, _type_name = obj.get_type("")
     if is_array:
@@ -52,7 +52,7 @@ def _to_python(obj: Any) -> Any:
     return obj.get_val_string("", 0)
 
 
-def dict_to_property_object(engine: Engine, data: dict[str, Any], container: Any = None) -> Any:
+def dict_to_property_object(engine: Engine, data: dict[str, Any], container: Any = None) -> Any:  # noqa: ANN401
     """Rebuild a container PropertyObject from a dict produced by the walk above."""
     if container is None:
         container = engine.new_property_object(PropValType.Container, False, "", 0)
@@ -61,7 +61,7 @@ def dict_to_property_object(engine: Engine, data: dict[str, Any], container: Any
     return container
 
 
-def _set_member(engine: Engine, container: Any, name: str, value: Any) -> None:
+def _set_member(engine: Engine, container: Any, name: str, value: Any) -> None:  # noqa: ANN401
     """Create one sub-property (container, array, or scalar) from a Python value."""
     if isinstance(value, dict):
         container.new_sub_property(name, PropValType.Container, False, "", INSERT_IF_MISSING)
@@ -101,14 +101,14 @@ def _element_type(items: list[Any]) -> PropValType:
     return PropValType.String
 
 
-def _child(container: Any, name: str) -> Any:
+def _child(container: Any, name: str) -> Any:  # noqa: ANN401
     """Fetch a sub-property object, asserting it exists (keeps type-checkers happy)."""
     obj = container.get_property_object(name, 0)
     assert obj is not None
     return obj
 
 
-def _build_example_container(engine: Engine) -> Any:
+def _build_example_container(engine: Engine) -> Any:  # noqa: ANN401
     """Scalars, a scalar array, a nested DigitalMultimeter container, and an array of them."""
     data = engine.new_property_object(PropValType.Container, False, "", 0)
     data.set_val_number("TestResult", INSERT_IF_MISSING, 0.0)
@@ -138,7 +138,7 @@ def _build_example_container(engine: Engine) -> Any:
     return data
 
 
-def _child_by_offset(array: Any, offset: int) -> Any:
+def _child_by_offset(array: Any, offset: int) -> Any:  # noqa: ANN401
     """Fetch an array element object, asserting it exists (keeps type-checkers happy)."""
     element = array.get_property_object_by_offset(offset, 0)
     assert element is not None

{py_teststand-0.2.2 → py_teststand-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "py-teststand"
-version = "0.2.2"
+version = "0.2.3"
 description = "Community object-oriented Python 3 bindings for the National Instruments TestStand™ COM API"
 readme = "README.md"
 # Last supported by uv 0.11.15 (3cffe97c2 2026-05-18 x86_64-pc-windows-msvc)
@@ -132,6 +132,12 @@ url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
 explicit = true
 
+[tool.pyrefly]
+project-includes = [
+    "**/*.py*",
+    "**/*.ipynb",
+]
+
 [[tool.uv.index]]
 name = "pypi"
 url = "https://pypi.org/simple"

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/__init__.py

@@ -55,7 +55,9 @@ from .core.com_wrapper import COMWrapper, ts_interface  # noqa: F401
 from .core.engine import (
     AdapterKeyName,
     AllowAutomaticTypeConflictResolution,
+    ConflictHandler,
     ConflictResolution,
+    DefaultModelCallback,
     EditKind,
     Engine,
     FindFilePromptOption,
@@ -153,7 +155,7 @@ from .property.property_object_file import (
 from .sequence.code_template import CodeTemplate, CodeTemplates
 from .sequence.expression import EvaluationType, Expression
 from .sequence.location import AutoCreateVariableLocation, Location, Locations
-from .sequence.sequence import Sequence
+from .sequence.sequence import Sequence, SequenceFileType, SequenceType
 from .sequence.sequence_context import SequenceContext
 from .sequence.sequence_file import SequenceFile
 from .sequence.step import RunMode, Step
@@ -209,12 +211,14 @@ __all__ = [
     "CheckedState",
     "CodeTemplate",
     "CodeTemplates",
+    "ConflictHandler",
     "ConflictResolution",
     "DLLAdapter",
     "DLLModule",
     "DataType",
     "DatabaseLogOptions",
     "DebugOption",
+    "DefaultModelCallback",
     "DeploymentError",
     "DotNetAdapter",
     "DotNetModule",
@@ -314,8 +318,10 @@ __all__ = [
     "SequenceContext",
     "SequenceFile",
     "SequenceFileLoadError",
+    "SequenceFileType",
     "SequenceFileViewManager",
     "SequenceTerminatedError",
+    "SequenceType",
     "SequenceValidationError",
     "SourceControlCommand",
     "SourceControlCommandOption",

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/adapters/adapter.py

@@ -97,6 +97,10 @@ class Adapter(PropertyObject):
     def name(self) -> str:
         return self.key_name
 
+    @name.setter
+    def name(self, value: str) -> None:  # noqa: ARG002
+        raise AttributeError("name is read-only on Adapter")
+
     @property
     @ts_interface
     def key_name(self) -> str:
@@ -525,16 +529,6 @@ class Module(PropertyObject):
 
         return SequenceCallModule(self._com_obj, self._engine_ref)
 
-    @property
-    @ts_interface
-    def comment(self) -> str:
-        return str(self._com_obj.Comment)
-
-    @property
-    @ts_interface
-    def name(self) -> str:
-        return str(self._com_obj.Name)
-
     @property
     @ts_interface
     def description(self) -> str:

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/adapters/dotnet.py

@@ -908,8 +908,9 @@ class DotNetParameter(PropertyObject):
         return DotNetParameters(self._com_obj.Elements, self._engine_ref)
 
     @ts_interface
-    def display_create_custom_data_type_dialog(self, sequence_context: bool) -> typing.Any:  # noqa: ANN401
-        return bool(self._com_obj.DisplayCreateCustomDataTypeDialog(sequence_context._com_obj))
+    def display_create_custom_data_type_dialog(self, sequence_context: typing.Any) -> typing.Any:  # noqa: ANN401
+        raw_ctx = getattr(sequence_context, "_com_obj", sequence_context)
+        return bool(self._com_obj.DisplayCreateCustomDataTypeDialog(raw_ctx))
 
     @property
     @ts_interface
@@ -960,6 +961,11 @@ class DotNetParameter(PropertyObject):
     def name(self) -> str:
         return str(self._com_obj.Name)
 
+    @name.setter
+    @ts_interface
+    def name(self, value: str) -> None:  # noqa: ARG002
+        raise AttributeError("name is read-only on DotNetParameter")
+
     @property
     @ts_interface
     def parameter_name(self) -> str:

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/adapters/labview.py

@@ -1050,6 +1050,11 @@ class LabVIEWParameter(PropertyObject):
     def array_element_prototype(self) -> typing.Any:  # noqa: ANN401
         return self._com_obj.ArrayElementPrototype
 
+    @array_element_prototype.setter
+    @ts_interface
+    def array_element_prototype(self, value: typing.Any) -> None:  # noqa: ANN401, ARG002
+        raise AttributeError("array_element_prototype is read-only")
+
     @ts_interface
     def as_property_object(self) -> PropertyObject:
         return PropertyObject(self._com_obj.AsPropertyObject(), self._engine_ref)
@@ -1258,6 +1263,11 @@ class LabVIEWParameterElement(PropertyObject):
     def array_element_prototype(self) -> typing.Any:  # noqa: ANN401
         return self._com_obj.ArrayElementPrototype
 
+    @array_element_prototype.setter
+    @ts_interface
+    def array_element_prototype(self, value: typing.Any) -> None:  # noqa: ANN401, ARG002
+        raise AttributeError("array_element_prototype is read-only")
+
     @ts_interface
     def as_property_object(self) -> PropertyObject:
         return PropertyObject(self._com_obj.AsPropertyObject(), self._engine_ref)

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/adapters/labview_nxg.py

@@ -348,7 +348,7 @@ class LabVIEWNXGModule(Module):
         self._com_obj.GLLPath = str(value)
 
     @ts_interface
-    def have_properties_changed(self, options: int = 0) -> bool:
+    def have_properties_changed(self, options: int = 0) -> tuple[bool, list[typing.Any]]:
         result = self._com_obj.HavePropertiesChanged(None, int(options))
         if isinstance(result, tuple) and len(result) >= 2:
             return bool(result[0]), list(result[1])
@@ -458,6 +458,11 @@ class LabVIEWNXGProject(PropertyObject):
     def name(self) -> str:
         return str(self._com_obj.Name)
 
+    @name.setter
+    @ts_interface
+    def name(self, value: str) -> None:  # noqa: ARG002
+        raise AttributeError("name is read-only on LabVIEWNXGProject")
+
 
 class LabVIEWNXGProjectItems(PropertyObject):
     @property
@@ -640,6 +645,11 @@ class LabVIEWNXGParameter(PropertyObject):
     def array_element_prototype(self) -> LabVIEWNXGParameter:
         return LabVIEWNXGParameter(self._com_obj.ArrayElementPrototype, self._engine_ref)
 
+    @array_element_prototype.setter
+    @ts_interface
+    def array_element_prototype(self, value: typing.Any) -> None:  # noqa: ANN401, ARG002
+        raise AttributeError("array_element_prototype is read-only")
+
     @ts_interface
     def as_property_object(self) -> PropertyObject:
         return PropertyObject(self._com_obj.AsPropertyObject(), self._engine_ref)

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/adapters/sequence.py

@@ -305,6 +305,11 @@ class SequenceCallParameter(PropertyObject):
     def name(self) -> str:
         return str(self._com_obj.Name)
 
+    @name.setter
+    @ts_interface
+    def name(self, value: str) -> None:  # noqa: ARG002
+        raise AttributeError("name is read-only on SequenceCallParameter")
+
     @property
     @ts_interface
     def type(self) -> int:

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/analyzer/rule_configuration.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import typing
+
 from py_teststand.analyzer.rule import RuleSeverity
 from py_teststand.analyzer.rule_setting_values import RuleSettingValues
 from py_teststand.core.com_wrapper import ts_interface
@@ -56,8 +58,10 @@ class RuleConfiguration(PropertyObject):
 class RuleConfigurationContext(PropertyObject):
     @property
     @ts_interface
-    def engine(self) -> PropertyObject:
-        return PropertyObject(self._com_obj.Engine, self._engine_ref)
+    def engine(self) -> typing.Any:  # noqa: ANN401
+        from py_teststand.core.engine.engine import Engine
+
+        return Engine(self._com_obj.Engine)
 
     @property
     @ts_interface

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/core/com_wrapper.py

@@ -122,6 +122,7 @@ def ts_interface(func: T) -> T:
 
 class COMWrapper:
     _com_obj: COM
+    _engine_ref: typing.Callable[[], Engine | typing.Any | None]
 
     def __init__(self, com_obj: COM, engine: Engine | typing.Any | None = None) -> None:  # noqa: ANN401
 

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/core/engine/engine.py

@@ -789,6 +789,8 @@ class Engine(COMWrapper):
             int(dlg_options),
             selected_item_com,
         )
+        from py_teststand.execution.breakpoint import SelectedBreakpointItem
+
         item = SelectedBreakpointItem(selected_item_com, self) if selected_item_com else None
         return bool(result), item
 
@@ -1700,7 +1702,7 @@ class Engine(COMWrapper):
         return OutputMessages(self._engine.GetOutputMessages(), self)
 
     @ts_interface
-    def get_product_registration_info(self) -> bool:
+    def get_product_registration_info(self) -> tuple[bool, str, str, str]:
         user_out = ""
         company_out = ""
         serial_out = ""
@@ -1845,7 +1847,7 @@ class Engine(COMWrapper):
         )
 
     @ts_interface
-    def get_tool_menu_item_info_with_id(self, unique_item_id: int) -> str:
+    def get_tool_menu_item_info_with_id(self, unique_item_id: int) -> tuple[str, int, bool]:
         item_text_out = ""
         sub_menu_index_out = 0
         enabled_out = False

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/execution/sync_manager.py

@@ -167,7 +167,7 @@ class Mutex(SyncObject):
         self._com_obj.EarlyUnlockMutexEx(com_context, thread_id)
 
     @ts_interface
-    def get_saved_owner_info(self) -> str:
+    def get_saved_owner_info(self) -> tuple[str, str, SyncManager]:
         res = self._com_obj.GetSavedOwnerInfo()
         return str(res[0]), str(res[1]), SyncManager(res[2], self._engine_ref)
 
@@ -180,7 +180,7 @@ class Mutex(SyncObject):
         return int(self._com_obj.GetInfo())
 
     @ts_interface
-    def get_info_ex(self) -> int:
+    def get_info_ex(self) -> tuple[int, int, str]:
         res = self._com_obj.GetInfoEx()
         return int(res[0]), int(res[1]), str(res[2])
 
@@ -216,7 +216,7 @@ class Semaphore(SyncObject):
         return AutoReleaser(res[0], self._engine_ref), WaitResult(res[1])
 
     @ts_interface
-    def get_info(self) -> int:
+    def get_info(self) -> tuple[int, int, int, int]:
         res = self._com_obj.GetInfo()
         return int(res[0]), int(res[1]), int(res[2]), int(res[3])
 
@@ -262,7 +262,7 @@ class Batch(SyncObject):
         return [Thread(t, self._engine_ref) for t in com_threads]
 
     @ts_interface
-    def get_info(self) -> int:
+    def get_info(self) -> tuple[int, int]:
         res = self._com_obj.GetInfo()
         return int(res[0]), int(res[1])
 
@@ -377,7 +377,7 @@ class Notification(SyncObject):
         self._com_obj.ClearEx(com_context)
 
     @ts_interface
-    def get_info(self) -> int:
+    def get_info(self) -> tuple[int, bool]:
         res = self._com_obj.GetInfo()
         return int(res[0]), bool(res[1])
 
@@ -453,7 +453,7 @@ class Notification(SyncObject):
 
 class Rendezvous(SyncObject):
     @ts_interface
-    def get_info(self) -> int:
+    def get_info(self) -> tuple[int, int]:
         res = self._com_obj.GetInfo()
         return int(res[0]), int(res[1])
 
@@ -531,7 +531,7 @@ class Queue(SyncObject):
         return list(self._com_obj.Flush(com_context))
 
     @ts_interface
-    def get_info(self) -> int:
+    def get_info(self) -> tuple[int, int, int]:
         res = self._com_obj.GetInfo()
         return int(res[0]), int(res[1]), int(res[2])
 
@@ -729,12 +729,12 @@ class SyncManager(COMWrapper):
         return SyncObject(res, self._engine_ref)
 
     @ts_interface
-    def get_mutex_owner_info(self, mutex_name: str) -> str:
+    def get_mutex_owner_info(self, mutex_name: str) -> tuple[str, str, SyncManager]:
         res = self._com_obj.GetMutexOwnerInfo(mutex_name)
         return str(res[0]), str(res[1]), SyncManager(res[2], self._engine_ref)
 
     @ts_interface
-    def get_lock_mutex_thread_is_waiting_for(self, thread_id: str) -> str:
+    def get_lock_mutex_thread_is_waiting_for(self, thread_id: str) -> tuple[str, SyncManager]:
         res = self._com_obj.GetLockMutexThreadIsWaitingFor(thread_id)
         return str(res[0]), SyncManager(res[1], self._engine_ref)
 

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/property/property_object.py

@@ -387,7 +387,7 @@ class PropertyObject(COMWrapper):
                 except Exception:
                     value = tuple(typing.cast("typing.Any", value))
         elif isinstance(value, datetime.datetime) and pywintypes is not None:
-            value = pywintypes.Time(value)
+            value = pywintypes.Time(typing.cast("typing.Any", value))
         self._property_object.SetValVariant(lookup_string, options, value)
 
     @ts_interface
@@ -533,7 +533,7 @@ class PropertyObject(COMWrapper):
         return bool(self._property_object.Exists(lookup_string, options))
 
     @ts_interface
-    def validate_new_name(self, new_name: str, allow_duplicates: bool = False) -> bool:
+    def validate_new_name(self, new_name: str, allow_duplicates: bool = False) -> tuple[bool, str]:
         res = self._property_object.ValidateNewName(new_name, allow_duplicates)
         if isinstance(res, tuple) and len(res) >= 2:
             return bool(res[1]), str(res[0])
@@ -553,13 +553,12 @@ class PropertyObject(COMWrapper):
     @property
     @ts_interface
     def name(self) -> str:
-        return str(getattr(self._com_obj, "Name", ""))
+        return str(self._property_object.Name)
 
     @name.setter
     @ts_interface
     def name(self, value: str) -> None:
-        if hasattr(self._com_obj, "Name"):
-            self._com_obj.Name = value
+        self._property_object.Name = value
 
     @functools.cached_property
     @ts_interface
@@ -837,7 +836,7 @@ class PropertyObject(COMWrapper):
                 except Exception:
                     value = value
         elif isinstance(value, datetime.datetime) and pywintypes is not None:
-            value = pywintypes.Time(value)
+            value = pywintypes.Time(typing.cast("typing.Any", value))
         self._property_object.SetValVariantByOffset(array_offset, options, value)
 
     @ts_interface

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/sequence/code_template.py

@@ -31,6 +31,11 @@ class CodeTemplate(PropertyObject):
     def name(self) -> str:
         return str(self._com_obj.Name)
 
+    @name.setter
+    @ts_interface
+    def name(self, value: str) -> None:  # noqa: ARG002
+        raise AttributeError("name is read-only on CodeTemplate")
+
     @property
     @ts_interface
     def description(self) -> str:

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/sequence/sequence_file.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 import functools
 import typing
-import weakref
 from enum import Enum, IntEnum
 from typing import TYPE_CHECKING
 
@@ -112,12 +111,7 @@ class SequenceCollection:
 class SequenceFile(PropertyObjectFile):
     def __init__(self, com_obj: typing.Any, engine: Engine | typing.Any | None = None) -> None:  # noqa: ANN401
 
-        super().__init__(com_obj)
-        self._engine_ref = (
-            engine
-            if isinstance(engine, weakref.ReferenceType) or engine is None
-            else weakref.ref(engine)
-        )
+        super().__init__(com_obj, engine)
 
     def __enter__(self) -> SequenceFile:
 
@@ -211,7 +205,12 @@ class SequenceFile(PropertyObjectFile):
     @property
     @ts_interface
     def change_count(self) -> int:
-        return int(self._com_obj.ChangeCount)
+        return int(self.as_property_object_file().change_count)
+
+    @change_count.setter
+    @ts_interface
+    def change_count(self, value: int) -> None:
+        self.as_property_object_file().change_count = value
 
     @property
     @ts_interface
@@ -367,7 +366,11 @@ class SequenceFile(PropertyObjectFile):
         )
 
     @ts_interface
-    def load_modules(self, options: int = 0, context: bool | None = None) -> typing.Any:  # noqa: ANN401
+    def load_modules(
+        self,
+        options: int = 0,
+        context: typing.Any | None = None,  # noqa: ANN401
+    ) -> typing.Any:  # noqa: ANN401
         ctx_com = getattr(context, "_com_obj", context)
         return bool(self._com_obj.LoadModules(int(options), ctx_com))
 

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/sequence/step_type.py

@@ -27,7 +27,6 @@ class StepType(PropertyObject):
 
         super().__init__(com_obj, engine_ref)
 
-    @property
     @ts_interface
     def as_property_object(self) -> PropertyObject:
         return PropertyObject(self._com_obj, self._engine_ref)

{py_teststand-0.2.2 → py_teststand-0.2.3}/src/py_teststand/ui/events.py

@@ -148,7 +148,7 @@ class ApplicationMgrEventsSink(UIEventSink):
 
     def on_query_reload_sequence_file(
         self,
-        file: typing.Any,
+        file: typing.Any,  # noqa: ANN401
         reload_option: typing.Any,  # noqa: ANN401
     ) -> typing.Any:  # noqa: ANN401
         try:
@@ -222,8 +222,8 @@ class ApplicationMgrEventsSink(UIEventSink):
 
     def on_query_close_execution(
         self,
-        execution: typing.Any,
-        run_state: typing.Any,
+        execution: typing.Any,  # noqa: ANN401
+        run_state: typing.Any,  # noqa: ANN401
         option: typing.Any,  # noqa: ANN401
     ) -> typing.Any:  # noqa: ANN401
         try:
@@ -359,8 +359,8 @@ class ApplicationMgrEventsSink(UIEventSink):
 
     def on_break(
         self,
-        execution: typing.Any,
-        thread: typing.Any,
+        execution: typing.Any,  # noqa: ANN401
+        thread: typing.Any,  # noqa: ANN401
         sequence_context: typing.Any,  # noqa: ANN401
     ) -> typing.Any:  # noqa: ANN401
         try:
@@ -496,8 +496,8 @@ class ExecutionViewMgrEventsSink(UIEventSink):
 
     def on_break(
         self,
-        execution: typing.Any,
-        thread: typing.Any,
+        execution: typing.Any,  # noqa: ANN401
+        thread: typing.Any,  # noqa: ANN401
         sequence_context: typing.Any,  # noqa: ANN401
     ) -> typing.Any:  # noqa: ANN401
         try:
@@ -507,8 +507,8 @@ class ExecutionViewMgrEventsSink(UIEventSink):
 
     def on_trace(
         self,
-        execution: typing.Any,
-        thread: typing.Any,
+        execution: typing.Any,  # noqa: ANN401
+        thread: typing.Any,  # noqa: ANN401
         sequence_context: typing.Any,  # noqa: ANN401
     ) -> typing.Any:  # noqa: ANN401
         try:

py_teststand-0.2.3/tests/unit/test_public_api.py

@@ -0,0 +1,52 @@
+import importlib
+import inspect
+import pkgutil
+import warnings
+from enum import Enum
+
+import py_teststand
+from py_teststand.core.com_wrapper import COMWrapper
+
+
+def test_public_api_exports_all_classes() -> None:
+    """Ensure that all public COM wrappers and Enums are exported in the root __init__.py."""
+    missing_exports: list[str] = []
+
+    package = py_teststand
+    prefix = package.__name__ + "."
+
+    for _importer, modname, _ispkg in pkgutil.walk_packages(package.__path__, prefix):
+        if "tests" in modname or "conftest" in modname:
+            continue
+
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            try:
+                module = importlib.import_module(modname)
+            except Exception:
+                continue
+
+        for name, obj in inspect.getmembers(module):
+            if name.startswith("_"):
+                continue
+
+            # Only consider classes defined in py_teststand (not imported from outside)
+            if inspect.isclass(obj) and getattr(obj, "__module__", "").startswith("py_teststand"):
+                if issubclass(obj, (COMWrapper, Enum)):
+                    if name not in py_teststand.__all__:
+                        missing_exports.append(f"{name} (from {obj.__module__})")
+
+    missing_exports = sorted(set(missing_exports))
+
+    # Exceptions that are purposefully not in __all__
+    # (add to this list if there are actual internal classes that shouldn't be exposed)
+    allowed_missing = {
+        "BaseAdapter (from py_teststand.adapters.adapter)",
+        "UIMessageHandler (from py_teststand.ext.events)",
+    }
+
+    missing_exports = [m for m in missing_exports if m not in allowed_missing]
+
+    assert not missing_exports, (
+        f"Missing exports in py_teststand.__init__.__all__: {missing_exports}"
+    )