move to pytest

Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>

Author: vikrantpuppala

tests/unit/test_thrift_field_ids.py

@@ -7,13 +7,17 @@
 
 import inspect
 import pytest
-import unittest
 
 from databricks.sql.thrift_api.TCLIService import ttypes
 
 
-class TestThriftFieldIds(unittest.TestCase):
-    """Test suite for validating Thrift field ID constraints."""
+class TestThriftFieldIds:
+    """
+    Unit test to validate that all Thrift-generated field IDs comply with the maximum limit.
+
+    Field IDs in Thrift must stay below 3329 to avoid conflicts with reserved ranges
+    and ensure compatibility with various Thrift implementations and protocols.
+    """
 
     MAX_ALLOWED_FIELD_ID = 3329
 
@@ -42,7 +46,7 @@ def test_all_thrift_field_ids_are_within_allowed_range(self):
 
         if violations:
             error_message = self._build_error_message(violations)
-            self.fail(error_message)
+            pytest.fail(error_message)
 
     def _check_class_field_ids(self, cls, class_name, violations):
         """