40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""Quick import test for new strategies."""
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# Add backend to path
|
|
backend_path = Path(__file__).parent
|
|
sys.path.insert(0, str(backend_path))
|
|
|
|
try:
|
|
from app.strategies.factors.value import ValueStrategy
|
|
from app.strategies.factors.quality import QualityStrategy
|
|
from app.strategies.factors.all_value import AllValueStrategy
|
|
from app.strategies.registry import STRATEGY_REGISTRY
|
|
|
|
print("✓ All imports successful")
|
|
print(f"✓ ValueStrategy: {ValueStrategy}")
|
|
print(f"✓ QualityStrategy: {QualityStrategy}")
|
|
print(f"✓ AllValueStrategy: {AllValueStrategy}")
|
|
print(f"\nRegistry contains {len(STRATEGY_REGISTRY)} strategies:")
|
|
for name in sorted(STRATEGY_REGISTRY.keys()):
|
|
print(f" - {name}")
|
|
|
|
# Test instantiation
|
|
value_strat = ValueStrategy(config={"count": 20})
|
|
quality_strat = QualityStrategy(config={"count": 20})
|
|
all_value_strat = AllValueStrategy(config={"count": 20})
|
|
|
|
print("\n✓ All strategies instantiated successfully")
|
|
print(f" - ValueStrategy.name: {value_strat.name}")
|
|
print(f" - QualityStrategy.name: {quality_strat.name}")
|
|
print(f" - AllValueStrategy.name: {all_value_strat.name}")
|
|
|
|
except Exception as e:
|
|
print(f"✗ Import failed: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
sys.exit(1)
|
|
|
|
print("\n✓ All tests passed!")
|