Code import - branch release/SIENTIAPDE-1645

2026-08-05 13:53:37 +00:00
commit d481e0acff
116 changed files with 92848 additions and 0 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -0,0 +1,103 @@
+"""
+Test bootstrap: stub optional `sientia_do` submodules not shipped in minimal installs.
+
+Must run before importing `model_manager.sientia.models` (pulled in via TrainModelParams).
+Stubs Evidently submodules so `model_manager.sientia.reports` imports (via DataManagerRepository).
+"""
+
+from __future__ import annotations
+
+import sys
+from types import ModuleType
+
+
+def _make_dummy(name: str) -> type:
+    return type(name, (), {})
+
+
+def _stub_evidently() -> None:
+    """Minimal Evidently API surface required to import `model_manager.sientia.reports`."""
+    ev = ModuleType('evidently')
+    sys.modules['evidently'] = ev
+
+    mp = ModuleType('evidently.metric_preset')
+    mp.DataDriftPreset = _make_dummy('DataDriftPreset')  # type: ignore[attr-defined]
+    sys.modules['evidently.metric_preset'] = mp
+
+    metrics = ModuleType('evidently.metrics')
+    _metric_names = (
+        'ColumnSummaryMetric',
+        'ConflictTargetMetric',
+        'DatasetCorrelationsMetric',
+        'DatasetSummaryMetric',
+        'RegressionAbsPercentageErrorPlot',
+        'RegressionDummyMetric',
+        'RegressionErrorDistribution',
+        'RegressionErrorPlot',
+        'RegressionPerformanceMetrics',
+        'RegressionPredictedVsActualPlot',
+        'RegressionPredictedVsActualScatter',
+    )
+    for n in _metric_names:
+        setattr(metrics, n, _make_dummy(n))
+    sys.modules['evidently.metrics'] = metrics
+
+    base = ModuleType('evidently.metrics.base_metric')
+
+    def generate_column_metrics(*_a, **_k):
+        return []
+
+    base.generate_column_metrics = generate_column_metrics  # type: ignore[attr-defined]
+    sys.modules['evidently.metrics.base_metric'] = base
+
+    opt = ModuleType('evidently.options')
+    opt.ColorOptions = _make_dummy('ColorOptions')  # type: ignore[attr-defined]
+    sys.modules['evidently.options'] = opt
+
+    pipeline = ModuleType('evidently.pipeline')
+    sys.modules['evidently.pipeline'] = pipeline
+
+    colmap = ModuleType('evidently.pipeline.column_mapping')
+    colmap.ColumnMapping = _make_dummy('ColumnMapping')  # type: ignore[attr-defined]
+    sys.modules['evidently.pipeline.column_mapping'] = colmap
+
+    rep = ModuleType('evidently.report')
+    rep.Report = _make_dummy('Report')  # type: ignore[attr-defined]
+    sys.modules['evidently.report'] = rep
+
+
+def pytest_configure(config) -> None:  # noqa: ARG001
+    """Register stub modules so imports used by production code resolve in CI/dev venvs."""
+    _stub_evidently()
+
+    if 'sientia_do.operations.df_preprocessor' not in sys.modules:
+        df_pre = ModuleType('sientia_do.operations.df_preprocessor')
+
+        def create_features(input_data, *_a, **_k):
+            return input_data
+
+        def limit_dataset(input_data, low_lim, upp_lim, *_a, **_k):
+            return input_data, low_lim, upp_lim
+
+        def treat_nan(input_data, *_a, **_k):
+            return input_data
+
+        df_pre.create_features = create_features  # type: ignore[attr-defined]
+        df_pre.limit_dataset = limit_dataset  # type: ignore[attr-defined]
+        df_pre.treat_nan = treat_nan  # type: ignore[attr-defined]
+        sys.modules['sientia_do.operations.df_preprocessor'] = df_pre
+
+    sys.modules.setdefault('sientia_do.operations', ModuleType('sientia_do.operations'))
+
+    if 'sientia_do.timeseries.analyzer' not in sys.modules:
+        ts_an = ModuleType('sientia_do.timeseries.analyzer')
+
+        class TimeSeriesDiscontinuityAnalyzer:  # noqa: D401
+            """Stub for tests."""
+
+            pass
+
+        ts_an.TimeSeriesDiscontinuityAnalyzer = TimeSeriesDiscontinuityAnalyzer  # type: ignore[attr-defined]
+        sys.modules['sientia_do.timeseries.analyzer'] = ts_an
+
+    sys.modules.setdefault('sientia_do.timeseries', ModuleType('sientia_do.timeseries'))