style: black

Author: jorwoods

tableauserverclient/config.py

@@ -9,15 +9,17 @@
 # The maximum size of a file that can be published in a single request is 64MB
 FILESIZE_LIMIT_MB = 64
 
+
 class Config:
-# For when a datasource is over 64MB, break it into 5MB(standard chunk size) chunks
+    # For when a datasource is over 64MB, break it into 5MB(standard chunk size) chunks
     @property
     def CHUNK_SIZE_MB(self):
         return int(os.getenv("TSC_CHUNK_SIZE_MB", 5 * 10))  # 5MB felt too slow, upped it to 50
 
-# Default page size
+    # Default page size
     @property
     def PAGE_SIZE(self):
         return int(os.getenv("TSC_PAGE_SIZE", 100))
 
+
 config = Config()

test/test_pager.py

@@ -13,6 +13,7 @@
 GET_XML_PAGE2 = os.path.join(TEST_ASSET_DIR, "workbook_get_page_2.xml")
 GET_XML_PAGE3 = os.path.join(TEST_ASSET_DIR, "workbook_get_page_3.xml")
 
+
 @contextlib.contextmanager
 def set_env(**environ):
     old_environ = dict(os.environ)
@@ -23,6 +24,7 @@ def set_env(**environ):
         os.environ.clear()
         os.environ.update(old_environ)
 
+
 class PagerTests(unittest.TestCase):
     def setUp(self):
         self.server = TSC.Server("http://test", False)
@@ -100,13 +102,13 @@ def test_pager_with_options(self):
             wb3 = workbooks.pop()
             self.assertEqual(wb3.name, "Page3Workbook")
 
-    def test_pager_with_env_var(self) -> None:
+    def test_pager_with_env_var(self):
         with set_env(TSC_PAGE_SIZE="1000"):
             assert config.PAGE_SIZE == 1000
             loop = TSC.Pager(self.server.workbooks)
             assert loop._options.pagesize == 1000
 
-    def test_queryset_with_env_var(self) -> None:
+    def test_queryset_with_env_var(self):
         with set_env(TSC_PAGE_SIZE="1000"):
             assert config.PAGE_SIZE == 1000
             loop = self.server.workbooks.all()