From 95fc4cf322b2141f76142e46e96e2846f5034edf Mon Sep 17 00:00:00 2001 From: Janine vN Date: Tue, 22 Jul 2025 15:43:02 -0400 Subject: Add dynamic module loading fix for dataclasses --- resources/unittest_template.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/resources/unittest_template.py b/resources/unittest_template.py index d1944ee..5cdb8ec 100644 --- a/resources/unittest_template.py +++ b/resources/unittest_template.py @@ -52,6 +52,12 @@ def _load_user_module(result_writer, exit_wrapper: _EXIT_WRAPPER_TYPE) -> Module exit_wrapper(5) _module = ModuleType("module") + # It's necessary to manually add the module to the sys modules + # Dataclasses do not add themselves which causes issues for us with this type of dynamic loading + # if from __future__ import annotations is also used + # See: https://github.com/mkdocs/mkdocs/issues/3141 and https://github.com/sqlalchemy/alembic/issues/1419 + sys.modules[_module.__name__] = _module + exec(code, _module.__dict__) return _module -- cgit v1.2.3