from DataOperations import *
from get_test_data import get_test_data
import numpy as np


# Code prints results for 50 matches from 3 different leagues
test_data = get_test_data()
team_a_win = test_data["team_a_win"]
teams_a = test_data["teams_a"]
teams_b = test_data["teams_b"]
team_a_games = test_data["team_a_games"]
team_b_games = test_data["team_b_games"]

FS = create_fuzzy_system()

for x in range(50):
    if x < 20:
        res = get_prediction(teams_a[x], teams_b[x], team_a_games[x], team_b_games[x], "lcs_summer_2020", FS, verbose=True)
    elif x < 35:
        res = get_prediction(teams_a[x], teams_b[x], team_a_games[x], team_b_games[x], "lec_summer_2021", FS, verbose=True)
    else:
        res = get_prediction(teams_a[x], teams_b[x], team_a_games[x], team_b_games[x], "ultraliga_summer_2022", FS, verbose=True)

    print(f"Prediction: {res}")