class dao_test(unittest.TestCase): def setUp(self): #testing reference http://www.zerozero.pt/edition.php?jornada_in=19&id_edicao=86785&fase=81592 self.dao = DAO(country="england", season="15-16", cols="sport_cols") def test_get_data(self): #each team plays 19 time as HomeTeam self.assertEqual(20 * 19, len(self.dao.get_data())) def test_half_league_data(self): half1 = self.dao.half_league_data(half=1) self.assertEqual(max(half1["Date"]), to_datetime("30/12/2015", dayfirst=True)) self.assertEqual(min(half1["Date"]), to_datetime("08/08/2015", dayfirst=True)) def test_search_percentil_date(self): p05 = self.dao.search_percentil_date(percentile=0.5) self.assertEqual(p05, to_datetime("30/12/2015", dayfirst=True)) p0316 = self.dao.search_percentil_date(percentile=0.316) self.assertEqual(p0316, to_datetime("08/11/2015", dayfirst=True)) p06316 = self.dao.search_percentil_date(percentile=0.6316) self.assertEqual(p06316, to_datetime("03/02/2016", dayfirst=True))
from dao.dao import DAO from grouped_stats import GroupedStats from table import Table from dates import next_day from analysis import Analysis, CORRELATION_H1_H2, CORRELATION_H1_FULL, CORRELATION_H2_FULL dao = DAO(country="england", season="15-16", cols="sport_cols") stats = GroupedStats(dao.get_data()) print("") table = Table(dao) points = table.points() print ("Final table") print (points) print() ### FIRST HALF LEAGUE half_league_date = table.half_league_date() print(half_league_date) eng_points_h1 = table.points(to_date=half_league_date) eng_points_h2 = table.points(from_date=next_day(half_league_date)) print("England First Half") print(eng_points_h1.sort_values(by="Points", ascending=False)) print() print("England Second Half") print(eng_points_h2.sort_values(by="Points", ascending=False)) print()