From 6e819533e41690d2ddfa27f65d2af1af3b33e5e2 Mon Sep 17 00:00:00 2001
From: Jordan Woods <13803242+jorwoods@users.noreply.github.com>
Date: Fri, 29 Aug 2025 22:05:23 -0500
Subject: [PATCH] chore: pytestify test_datasource_model

---
 test/test_datasource_model.py | 28 +++++++++++++++-------------
 1 file changed, 15 insertions(+), 13 deletions(-)

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