formatting, mypy

Author: jacalata

tableauserverclient/models/column_item.py

@@ -68,4 +68,4 @@ def _parse_element(column_xml, ns):
         description = column_xml.get("description", None)
         remote_type = column_xml.get("remoteType", None)
 
-        return id, name, description, remote_type
+        return id, name, description, remote_type

test/models/test_repr.py

@@ -9,6 +9,7 @@
 def is_concrete(obj: Any):
     return inspect.isclass(obj) and not inspect.isabstract(obj)
 
+
 @pytest.mark.parametrize("class_name, obj", inspect.getmembers(TSC, is_concrete))
 def test_by_reflection(class_name, obj):
     instance = try_instantiate_class(class_name, obj)
@@ -22,7 +23,6 @@ def test_by_reflection(class_name, obj):
             print(instance.__repr__.__name__)
 
 
-
 # Instantiate a class if it doesn't require any parameters
 def try_instantiate_class(name: str, obj: Any) -> Any | None:
     # Get the constructor (init) of the class

test/test_custom_view.py

@@ -175,6 +175,7 @@ def test_publish_filepath(server: TSC.Server) -> None:
     cv._owner = TSC.UserItem()
     cv._owner._id = "dd2239f6-ddf1-4107-981a-4cf94e415794"
     cv.workbook = TSC.WorkbookItem()
+    assert cv.workbook is not None
     cv.workbook._id = "1f951daf-4061-451a-9df1-69a8062664f2"
     with requests_mock.mock() as m:
         m.post(server.custom_views.expurl, status_code=201, text=GET_XML.read_text())
@@ -191,6 +192,7 @@ def test_publish_file_str(server: TSC.Server) -> None:
     cv._owner = TSC.UserItem()
     cv._owner._id = "dd2239f6-ddf1-4107-981a-4cf94e415794"
     cv.workbook = TSC.WorkbookItem()
+    assert cv.workbook is not None
     cv.workbook._id = "1f951daf-4061-451a-9df1-69a8062664f2"
     with requests_mock.mock() as m:
         m.post(server.custom_views.expurl, status_code=201, text=GET_XML.read_text())
@@ -207,6 +209,7 @@ def test_publish_file_io(server: TSC.Server) -> None:
     cv._owner = TSC.UserItem()
     cv._owner._id = "dd2239f6-ddf1-4107-981a-4cf94e415794"
     cv.workbook = TSC.WorkbookItem()
+    assert cv.workbook is not None
     cv.workbook._id = "1f951daf-4061-451a-9df1-69a8062664f2"
     data = io.BytesIO(CUSTOM_VIEW_DOWNLOAD.read_bytes())
     with requests_mock.mock() as m:
@@ -223,6 +226,7 @@ def test_publish_missing_owner_id(server: TSC.Server) -> None:
     cv = TSC.CustomViewItem(name="test")
     cv._owner = TSC.UserItem()
     cv.workbook = TSC.WorkbookItem()
+    assert cv.workbook is not None
     cv.workbook._id = "1f951daf-4061-451a-9df1-69a8062664f2"
     with requests_mock.mock() as m:
         m.post(server.custom_views.expurl, status_code=201, text=GET_XML.read_text())
@@ -244,8 +248,10 @@ def test_publish_missing_wb_id(server: TSC.Server) -> None:
 def test_large_publish(server: TSC.Server):
     cv = TSC.CustomViewItem(name="test")
     cv._owner = TSC.UserItem()
+    assert cv.workbook is not None
     cv._owner._id = "dd2239f6-ddf1-4107-981a-4cf94e415794"
     cv.workbook = TSC.WorkbookItem()
+    assert cv.workbook is not None
     cv.workbook._id = "1f951daf-4061-451a-9df1-69a8062664f2"
     with ExitStack() as stack:
         temp_dir = stack.enter_context(TemporaryDirectory())