Upload tests/test_pipeline.py

tests/test_pipeline.py

@@ -25,7 +25,7 @@ class TestProvenTemplates:
             "test_field", "test", 1.0, 0, "Test field", "Test",
             SignConvention.LONG_HIGH, DatasetTier.TIER1
         )
-        expr = generate_alpha15_variant(field, group_key="subindustry")
+        expr = generate_alpha15_variant(field, group_key="subindustry", decay=5)
         assert "ts_decay_linear(" in expr
         assert "group_neutralize(" in expr
         assert "ts_rank(" in expr
@@ -37,10 +37,20 @@ class TestProvenTemplates:
             "test_field", "test", 1.0, 0, "Test field", "Test",
             SignConvention.LONG_LOW, DatasetTier.TIER1
         )
-        expr = generate_alpha15_variant(field, group_key="subindustry")
+        expr = generate_alpha15_variant(field, group_key="subindustry", decay=5)
         # LONG_LOW should prefix the field with minus
         assert "-zscore" in expr
         
+    def test_alpha15_custom_decay(self):
+        field = BrainField(
+            "test_field", "test", 1.0, 0, "Test field", "Test",
+            SignConvention.LONG_HIGH, DatasetTier.TIER1
+        )
+        expr = generate_alpha15_variant(field, group_key="subindustry", decay=10)
+        assert "ts_decay_linear(" in expr
+        assert ", 10)" in expr
+        assert ", 5)" not in expr
+        
     def test_batch_generation(self):
         batch = generate_batch_from_proven_templates(count=3)
         assert len(batch) <= 3
@@ -101,6 +111,16 @@ class TestLint:
         assert "pv13_customergraphrank_auth_rank" in FIELD_INDEX
         assert "pv13_customergraphrank_page_rank" in FIELD_INDEX
         
+    def test_mutate_vol_scale_with_uppercase_fields(self):
+        """Test vol_scale mutation handles uppercase field IDs (e.g., mdl77_2GlobalDev...)."""
+        # Use a realistic expression with uppercase in the field name
+        expr = "ts_decay_linear(group_neutralize(rank(ts_rank(mdl77_2GlobalDevField, 252)), subindustry), 5)"
+        from alpha_factory.deterministic.expression_mutator import mutate_vol_scale
+        variants = mutate_vol_scale(expr)
+        # Should produce at least one variant since the regex now handles [a-zA-Z]
+        assert len(variants) > 0, "vol_scale should match uppercase field IDs"
+        assert "ts_std(" in variants[0]["expression"], "vol_scale should wrap field with ts_std"
+        
     def test_no_typos_in_field_registry(self):
         """Verify no 'ustomer' typos remain (missing 'c' in customer)."""
         for fid in FIELD_INDEX:
@@ -256,10 +276,10 @@ class TestOperatorsCSV:
             assert op_name in csv_ops, f"Operator '{op_name}' in lint.py but not in operators.csv"
 
 
-@pytest.mark.asyncio
 class TestAsyncPipeline:
-    async def test_proven_template_pipeline(self):
+    def test_proven_template_pipeline(self):
         """Test running a batch in proven template mode."""
+        import asyncio
         from alpha_factory.orchestration.pipeline import AlphaPipeline
         config = load_config()
         config.batch_size = 3
@@ -267,7 +287,11 @@ class TestAsyncPipeline:
         config.enable_brain_client = False
         
         pipeline = AlphaPipeline(config)
-        result = await pipeline.run_batch(3)
+        
+        async def _run():
+            return await pipeline.run_batch(3)
+        
+        result = asyncio.run(_run())
         
         # Pipeline now has local sim + checklist gates — results may be killed
         assert "promoted" in result or "iterated" in result or "killed" in result
@@ -304,7 +328,22 @@ class TestThemeSampler:
         from alpha_factory.deterministic.theme_sampler import pick_theme
         existing = ["momentum", "value", "quality"]
         theme = pick_theme(existing, [], [])
-        assert theme not in existing or len(existing) >= 20  # May repeat if exhausted
+        # Theme should be from the defined THEME_FIELDS
+        assert theme in [
+            "earnings_surprise_momentum", "earnings_quality_signaling",
+            "asset_growth_anomaly", "forward_value_composite",
+            "liquidity_risk_premium", "multi_factor_momentum",
+            "news_reaction_drift", "analyst_guidance_revision",
+            "options_sentiment_pcr", "supply_chain_network",
+            "social_contrarian", "geographic_exposure",
+        ]
+        
+    def test_pick_theme_all_dead_returns_alive(self):
+        """If all themes are dead, pick_theme must still return a valid theme."""
+        from alpha_factory.deterministic.theme_sampler import pick_theme, THEME_FIELDS
+        all_themes = list(THEME_FIELDS.keys())
+        theme = pick_theme([], [], dead_themes=all_themes)
+        assert theme in THEME_FIELDS, "Should return a valid theme even when all are dead"
 
 
 class TestWQClient: