Skip to content

Commit a196281

Browse files
authored
chore: pytestify test_datasource_model (#1656)
1 parent e8aed24 commit a196281

File tree

1 file changed

+15
-13
lines changed

1 file changed

+15
-13
lines changed

test/test_datasource_model.py

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,20 @@
1-
import unittest
1+
import pytest
2+
23
import tableauserverclient as TSC
34

45

5-
class DatasourceModelTests(unittest.TestCase):
6-
def test_nullable_project_id(self):
7-
datasource = TSC.DatasourceItem(name="10")
8-
self.assertEqual(datasource.project_id, None)
6+
def test_nullable_project_id():
7+
datasource = TSC.DatasourceItem(name="10")
8+
assert datasource.project_id is None
9+
10+
11+
def test_require_boolean_flag_bridge_fail():
12+
datasource = TSC.DatasourceItem("10")
13+
with pytest.raises(ValueError):
14+
datasource.use_remote_query_agent = "yes"
915

10-
def test_require_boolean_flag_bridge_fail(self):
11-
datasource = TSC.DatasourceItem("10")
12-
with self.assertRaises(ValueError):
13-
datasource.use_remote_query_agent = "yes"
1416

15-
def test_require_boolean_flag_bridge_ok(self):
16-
datasource = TSC.DatasourceItem("10")
17-
datasource.use_remote_query_agent = True
18-
self.assertEqual(datasource.use_remote_query_agent, True)
17+
def test_require_boolean_flag_bridge_ok():
18+
datasource = TSC.DatasourceItem("10")
19+
datasource.use_remote_query_agent = True
20+
assert datasource.use_remote_query_agent

0 commit comments

Comments
 (0)