예제 #1
0
    def test_engine_works(self):
        engine = utils.connect_nfldb()
        test_query = (
            "SELECT play.description "
            "from play "
            "WHERE play.gsis_id = '2009080950' AND play.play_id=721;")

        plays_df = pd.read_sql(test_query, engine)

        assert (plays_df.iloc[0]['description'] ==
                '(6:55) L.White left guard for 3 yards, TOUCHDOWN.')
예제 #2
0
    def test_no_config_error(self):
        nfldb.db._config_home = "/boogaboogabooga"

        with pytest.raises(IOError):
            utils.connect_nfldb()