diff --git a/test/test_datasource.py b/test/test_datasource.py
index 7f4cca75..56eb11ab 100644
--- a/test/test_datasource.py
+++ b/test/test_datasource.py
@@ -895,7 +895,8 @@ def test_publish_description(server: TSC.Server) -> None:
     ds_elem = body.find(".//datasource")
     assert ds_elem is not None
     assert ds_elem.attrib["description"] == "Sample description"
-    
+
+
 def test_get_datasource_no_owner(server: TSC.Server) -> None:
     with requests_mock.mock() as m:
         m.get(server.datasources.baseurl, text=GET_NO_OWNER.read_text())
diff --git a/test/test_tableauauth_model.py b/test/test_tableauauth_model.py
index 195bcf0a..17db5277 100644
--- a/test/test_tableauauth_model.py
+++ b/test/test_tableauauth_model.py
@@ -1,12 +1,8 @@
-import unittest
+import pytest
 
 import tableauserverclient as TSC
 
 
-class TableauAuthModelTests(unittest.TestCase):
-    def setUp(self):
-        self.auth = TSC.TableauAuth("user", "password", site_id="site1", user_id_to_impersonate="admin")
-
-    def test_username_password_required(self):
-        with self.assertRaises(TypeError):
-            TSC.TableauAuth()
+def test_username_password_required():
+    with pytest.raises(TypeError):
+        TSC.TableauAuth()