chore: migrate to pytest

microsoft · Nov 8, 2023 · 3a1a016 · 3a1a016
1 parent 82058f9
commit 3a1a016
Show file tree

Hide file tree

Showing 4 changed files with 99 additions and 14 deletions.
diff --git a/src/python/poetry.lock b/src/python/poetry.lock
diff --git a/src/python/pyproject.toml b/src/python/pyproject.toml
@@ -11,6 +11,7 @@ azure-mgmt-datafactory = "^3.1.0"
 azure-identity = "^1.14.1"
 astor = "^0.8.1"
 docstring-parser = "^0.15"
+pytest = "^7.4.3"
 
 
 [build-system]

diff --git a/src/python/tests/__init__.py b/src/python/tests/__init__.py
diff --git a/src/python/tests/playground/example_test.py b/src/python/tests/playground/example_test.py
@@ -1,7 +1,4 @@
 import json
-import unittest
-from unittest import TestCase
-
 from azure.mgmt.datafactory import models as _models
 
 from data_factory_testing_framework.generated import Deserializer
@@ -23,23 +20,15 @@ def evaluate(activity: WebActivity):
     activity.body.evaluate()
 
 
-class TestPipeline(TestCase):
+class TestPipeline:
     pipeline: PipelineResource
 
-    @classmethod
-    def setUpClass(cls):
-        cls.pipeline = get_pipeline()
-
     def test_web_activity(self):
         # Arrange
-        web_activity: WebActivity = self.pipeline.activities[0]
+        web_activity: WebActivity = get_pipeline().activities[0]
 
         # Act
         evaluate(web_activity)
 
         # Assert
         assert web_activity.body.value == "test2"
-
-
-if __name__ == '__main__':
-    unittest.main()