magicciv/tools/sprite-generation/test_rank.py

29 lines
795 B
Python

#!/usr/bin/env python3
import asyncio
import sys
from pathlib import Path
sys.path.insert(0, 'engine')
from registry import SpriteRegistry
from ranker import ScoringPipeline
async def test_rank():
reg = SpriteRegistry(db_path=Path('spritegen.db'))
ranker = ScoringPipeline(registry=reg, raw_dir=Path('raw'))
sprite_id = 'units/bowmen_dwarves_f'
print(f"Starting rank for {sprite_id}...")
print(f"Getting sprite...")
sprite = reg.get_sprite(sprite_id)
print(f"Sprite: {sprite}")
print(f"Calling advance_sprite...")
try:
result = await ranker.advance_sprite(sprite_id)
print(f"Result: {result}")
except Exception as e:
print(f"Error: {e}")
import traceback
traceback.print_exc()
asyncio.run(test_rank())