diff --git a/test/test_datasource_model.py b/test/test_datasource_model.py
index 65528419..c74805fa 100644
--- a/test/test_datasource_model.py
+++ b/test/test_datasource_model.py
@@ -1,18 +1,20 @@
-import unittest
+import pytest
+
 import tableauserverclient as TSC
 
 
-class DatasourceModelTests(unittest.TestCase):
-    def test_nullable_project_id(self):
-        datasource = TSC.DatasourceItem(name="10")
-        self.assertEqual(datasource.project_id, None)
+def test_nullable_project_id():
+    datasource = TSC.DatasourceItem(name="10")
+    assert datasource.project_id is None
+
+
+def test_require_boolean_flag_bridge_fail():
+    datasource = TSC.DatasourceItem("10")
+    with pytest.raises(ValueError):
+        datasource.use_remote_query_agent = "yes"
 
-    def test_require_boolean_flag_bridge_fail(self):
-        datasource = TSC.DatasourceItem("10")
-        with self.assertRaises(ValueError):
-            datasource.use_remote_query_agent = "yes"
 
-    def test_require_boolean_flag_bridge_ok(self):
-        datasource = TSC.DatasourceItem("10")
-        datasource.use_remote_query_agent = True
-        self.assertEqual(datasource.use_remote_query_agent, True)
+def test_require_boolean_flag_bridge_ok():
+    datasource = TSC.DatasourceItem("10")
+    datasource.use_remote_query_agent = True
+    assert datasource.use_remote_query_agent