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()
