def setUp(self): config = ConfigParser() config.read(os.path.join(os.path.expanduser('~'), '.pgcred')) db = NBAPostgres(user=config['nbadb']['username'], password=config['nbadb']['password'], database=config['nbadb']['database']) self.a = RotoGuruNBAAgent(cache_name='test-rg', db=db) self.a.insert_db = False
def setUp(self): config = ConfigParser() config.read(os.path.join(os.path.expanduser('~'), '.pgcred')) self.db = NBAPostgres(user=config['nbadb']['username'], password=config['nbadb']['password'], database=config['nbadb']['database'])
def test_init(self): nbapg = NBAPostgres()
header = 'starting_lineups' _headers.append(header.lower()) return _headers if __name__ == "__main__": import glob import pprint import random from nba.db.pgsql import NBAPostgres logging.basicConfig(level=logging.ERROR) p = NBAStufferParser() db = NBAPostgres() fixed_games = [] for fn in glob.glob('/home/sansbacon/workspace/nba-data/nbastuffer/csv/*Box*.csv'): with open(fn, 'r') as infile: rows = infile.readlines() headers = rows.pop(0).split(',') headers = p.headers(headers) game_pairs = p.game_pairs(rows, headers) wanted = ['is_regular_season', 'has_ot', 'dataset', 'game_date', 'venue', 'rest_days', 'days_last_game', 'back_to_back', 'back_to_back_to_back', 'three_in_four', 'four_in_five', 'gamecode', 'team_code', 'team_id', 'opponent_team_code', 'opponent_team_id', 'q1', 'q2', 'q3', 'q4', 'ot1', 'ot2', 'ot3', 'ot4',
def setUp(self): self.db = NBAPostgres()