class NBAMongo_test(unittest.TestCase):
# changed NBAMongo class to require passing a db object
# this allows testing using ming in-memory instance
# still need a better way of having test data, probably should just pickle some examples and load them in tests
def setUp(self):
self.mg = ming.create_datastore('mim://', db='nba', **kwargs)
self.nbam = NBAMongo(self.mg.db)
self.games = []
self.standings = []
self._get_scoreboard()
def test_add_games(self):
ids = self.nbam.add_games(self.games)
self.assertGreater(ids.count, 0, "should have some ids after insert")
def test_get_games(self):
games = self.nbam.get_games()
self.assertGreater(games.count, 0, "should have some games after query")
'''
def test_get_games(self):
#,**kwargs):
pass
def test_add_standings(self):
#,documents,collection_name='standings'):
pass
def test_get_standings(self):
#,**kwargs):
pass
def test_add_player_boxscores(self):
#,documents,collection_name='player_boxscores'):
pass
def test_get_player_boxscores(self):
#,**kwargs):
pass
def test_add_team_boxscores(self):
#,documents,collection_name='team_boxscores'):
pass
def test_get_team_boxscores(self):
#,**kwargs):
pass
def test_add_playerstats(self):
#,documents,collection_name='playerstats'):
pass
def test_get_playerstats(self):
#,**kwargs):
pass
def test_add_players(self):
#,documents,collection_name='players'):
pass
def test_get_players(self):
#,**kwargs):
pass
def test_add_scoreboards(self):
#,documents,collection_name='scoreboards'):
pass
def test_get_scoreboards(self):
#,**kwargs):
pass
def test_add_player_gamelogs(self):
#,documents,collection_name='player_gamelogs'):
pass
def test_get_player_gamelogs(self):
#,**kwargs):
pass
def test_add_team_gamelogs(self):
#,documents,collection_name='team_gamelogs'):
pass
def test_get_team_gamelogs(self):
#,**kwargs):
pass
'''
def _get_scoreboard(self):
fn = '/home/sansbacon/workspace/nbacom-python/data/scoreboards/2015-01-24_scoreboard.json'
json_data = open(fn)
scoreboard = json.load(json_data)
for row_set in scoreboard['resultSets'][0]['rowSet']:
self.games.append(dict(zip(scoreboard['resultSets'][0]['headers'], row_set)))
for row_set in parsed['resultSets'][4]['rowSet']:
self.standings.append(dict(zip(parsed['resultSets'][4]['headers'], row_set)))
for row_set in parsed['resultSets'][5]['rowSet']:
standings.append(dict(zip(parsed['resultSets'][5]['headers'], row_set)))
return self.games, self.standings
def setUp(self):
self.mg = ming.create_datastore('mim://', db='nba', **kwargs)
self.nbam = NBAMongo(self.mg.db)
self.games = []
self.standings = []
self._get_scoreboard()
