from unittest import TestCase
import os
from sports_data.team import TeamInfoFactory

TEST_DATA_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'data')


class TestTeams(TestCase):
    def test_team_info(self):
        team_info = TeamInfoFactory()
        self.assertTrue(team_info.abbr_exists('LIV'))
        self.assertTrue(team_info.team_exists('Manchester United'))
        self.assertFalse(team_info.team_exists('Chicago Bears'))

        team_mu = team_info.get_team_from_abbr('MU')
        team_manchester_utd = team_info.get_team('Manchester Utd')
        self.assertEqual(team_mu, team_manchester_utd)

        with self.assertRaises(ValueError) as context:
            team_info.load_items_from_yaml(os.path.join(TEST_DATA_DIR, 'test_team_info.yaml'))
        self.assertTrue('duplicate abbreviations' in str(context.exception))