mirror of
https://github.com/mwisnowski/mtg_python_deckbuilder.git
synced 2025-09-22 04:50:46 +02:00
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Test Lightning Bolt directly"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'code'))
|
||
|
|
|
||
|
|
from deck_builder.include_exclude_utils import fuzzy_match_card_name
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
cards_df = pd.read_csv('csv_files/cards.csv', low_memory=False)
|
||
|
|
available_cards = set(cards_df['name'].dropna().unique())
|
||
|
|
|
||
|
|
# Test if Lightning Bolt gets the right score
|
||
|
|
result = fuzzy_match_card_name('bolt', available_cards)
|
||
|
|
print(f"'bolt' matches: {result.suggestions[:5]}")
|
||
|
|
|
||
|
|
result = fuzzy_match_card_name('lightn', available_cards)
|
||
|
|
print(f"'lightn' matches: {result.suggestions[:5]}")
|
||
|
|
|
||
|
|
# Check if Lightning Bolt is in the suggestions
|
||
|
|
if 'Lightning Bolt' in result.suggestions:
|
||
|
|
print(f"Lightning Bolt is suggestion #{result.suggestions.index('Lightning Bolt') + 1}")
|
||
|
|
else:
|
||
|
|
print("Lightning Bolt NOT in suggestions!")
|
||
|
|
|
||
|
|
# Test a few more obvious ones
|
||
|
|
result = fuzzy_match_card_name('lightning', available_cards)
|
||
|
|
print(f"'lightning' matches: {result.suggestions[:3]}")
|
||
|
|
|
||
|
|
result = fuzzy_match_card_name('warp', available_cards)
|
||
|
|
print(f"'warp' matches: {result.suggestions[:3]}")
|
||
|
|
|
||
|
|
# Also test the exact card name to make sure it's working
|
||
|
|
result = fuzzy_match_card_name('Lightning Bolt', available_cards)
|
||
|
|
print(f"'Lightning Bolt' exact: {result.matched_name} (confidence: {result.confidence:.3f})")
|