Minor bugfix

Author: Matteo Hoch

sdd_api/api.py

@@ -31,6 +31,8 @@ def __init__(self,username=None, password=None, client_id=None, client_secret=No
             raise ApiInitializationException("Please supply username, password, client_id, and client_secret params or pass a file")
         if server_url:
             self.SERVER_URL = server_url
+        else:
+            self.SERVER_URL = SERVER_URL
         self.verify=verify
         self.gen_token()
 
@@ -49,10 +51,8 @@ def get_json(self, table_name, schema_name="nfl", season_start=None, season_stop
         else:
             return d
 
-    def get_dataframe(self, table_name, schema_name="nfl", season_start=2017, season_stop=None, progress_bar=False):
+    def get_dataframe(self, table_name, schema_name="nfl", season_start=2017, season_stop=2017, progress_bar=True):
         df=pd.DataFrame()
-        if season_stop is None:
-            season_stop=2017
 
         seasons=range(season_start, season_stop+1)
         if progress_bar:

setup.py

@@ -10,7 +10,7 @@
 setup(
   name = 'sdd_api',
   packages = ['sdd_api'], # this must be the same as the name above
-  version = '0.2.0',
+  version = '0.2.1',
   description = 'Python API for Sports Data Direct',
   author = 'Sports Data Direct',
   author_email = 'admin@sportsdatadirect.com',

tests/critical_tests.py

@@ -14,11 +14,16 @@
 api = Api(username=username, password=password, client_id=client_id, client_secret=client_secret, server_url="https://localhost", verify=False)
 class TestConfig(unittest.TestCase):
     def test_config(self):
-        df = api.get_dataframe("drives")
-        assert type(df), pd.DataFrame
-    def test_players(self):
-        df = api.get_dataframe("player_game_logs", season_start=2011)
-        print(df.sample(10))
+        df = api.get_dataframe("matchups",season_start=2016)
+        assert len(df),len(df.drop_duplicates())
+    #     df = api.get_dataframe("drives")
+    #     assert type(df), pd.DataFrame
+    # def test_players(self):
+    #     df = api.get_dataframe("matchups", season_start=2016)
+    #     print(df.sample(10))
+    # def no_duplicates(self):
+    #     df = api.get_dataframe("matchups",season_start=2016)
+    #     assert len(df),len(df.drop_duplicates())
 
 if __name__ == '__main__':
     unittest.main()