(fix) Fixed pre-commit errors

Author: aarmoa

pyinjective/async_client.py

@@ -1617,7 +1617,9 @@ async def listen_derivative_market_updates(
     async def fetch_derivative_orderbook_v2(self, market_id: str, depth: Optional[int] = None) -> Dict[str, Any]:
         return await self.indexer_client.fetch_derivative_orderbook_v2(market_id=market_id, depth=depth or 0)
 
-    async def fetch_derivative_orderbooks_v2(self, market_ids: List[str], depth: Optional[int] = None) -> Dict[str, Any]:
+    async def fetch_derivative_orderbooks_v2(
+        self, market_ids: List[str], depth: Optional[int] = None
+    ) -> Dict[str, Any]:
         return await self.indexer_client.fetch_derivative_orderbooks_v2(market_ids=market_ids, depth=depth or 0)
 
     async def fetch_derivative_orders(

tests/test_async_client_deprecation_warnings.py

@@ -1,59 +1,57 @@
 import warnings
 from unittest.mock import MagicMock, patch
 
-import pytest
-
 from pyinjective.core.network import Network
 
 
 class TestAsyncClientDeprecationWarnings:
-    def test_async_client_deprecation_warning(self):
+    @patch("pyinjective.async_client.asyncio.get_event_loop")
+    @patch("pyinjective.async_client.ChainGrpcChainStream")
+    @patch("pyinjective.async_client.ChainGrpcExchangeApi")
+    @patch("pyinjective.async_client.ChainGrpcDistributionApi")
+    @patch("pyinjective.async_client.ChainGrpcAuthZApi")
+    @patch("pyinjective.async_client.ChainGrpcAuthApi")
+    @patch("pyinjective.async_client.ChainGrpcBankApi")
+    @patch("pyinjective.async_client.IndexerClient")
+    def test_async_client_deprecation_warning(self, *mocks):
         """Test that creating an AsyncClient instance raises a deprecation warning with correct details."""
-        with patch('pyinjective.async_client.IndexerClient'), \
-             patch('pyinjective.async_client.ChainGrpcBankApi'), \
-             patch('pyinjective.async_client.ChainGrpcAuthApi'), \
-             patch('pyinjective.async_client.ChainGrpcAuthZApi'), \
-             patch('pyinjective.async_client.ChainGrpcDistributionApi'), \
-             patch('pyinjective.async_client.ChainGrpcExchangeApi'), \
-             patch('pyinjective.async_client.ChainGrpcChainStream'), \
-             patch('pyinjective.async_client.asyncio.get_event_loop'):
-            
-            # Create a mock network to avoid actual network initialization
-            mock_network = MagicMock(spec=Network)
-            mock_network.chain_cookie_assistant = MagicMock()
-            mock_network.create_chain_grpc_channel = MagicMock()
-            mock_network.create_chain_stream_grpc_channel = MagicMock()
-            mock_network.official_tokens_list_url = "https://example.com/tokens.json"
-            
-            # Import here to avoid early import issues
-            from pyinjective.async_client import AsyncClient
-            
-            # Capture warnings
-            with warnings.catch_warnings(record=True) as warning_list:
-                warnings.simplefilter("always")  # Ensure all warnings are captured
-                
-                # Create AsyncClient instance - this should trigger the deprecation warning
-                client = AsyncClient(network=mock_network)
-                
-                # Find the AsyncClient deprecation warning
-                async_client_warnings = [
-                    w for w in warning_list 
-                    if issubclass(w.category, DeprecationWarning) and 
-                    "AsyncClient from pyinjective.async_client is deprecated" in str(w.message)
-                ]
-                
-                # Should have exactly one warning
-                assert len(async_client_warnings) == 1
-                
-                warning = async_client_warnings[0]
-                # Check warning message contains migration advice
-                assert "Please use AsyncClient from pyinjective.async_client_v2 instead" in str(warning.message)
-                # Check warning category
-                assert warning.category == DeprecationWarning
-                # Check stacklevel is working correctly (should point to this test file)
-                assert "test_async_client_deprecation_warnings.py" in warning.filename
-                
-                # Verify the client was still created successfully
-                assert client is not None
-                assert hasattr(client, 'network')
-                assert client.network == mock_network
+        # Create a mock network to avoid actual network initialization
+        mock_network = MagicMock(spec=Network)
+        mock_network.chain_cookie_assistant = MagicMock()
+        mock_network.create_chain_grpc_channel = MagicMock()
+        mock_network.create_chain_stream_grpc_channel = MagicMock()
+        mock_network.official_tokens_list_url = "https://example.com/tokens.json"
+
+        # Import here to avoid early import issues
+        from pyinjective.async_client import AsyncClient
+
+        # Capture warnings
+        with warnings.catch_warnings(record=True) as warning_list:
+            warnings.simplefilter("always")  # Ensure all warnings are captured
+
+            # Create AsyncClient instance - this should trigger the deprecation warning
+            client = AsyncClient(network=mock_network)
+
+            # Find the AsyncClient deprecation warning
+            async_client_warnings = [
+                w
+                for w in warning_list
+                if issubclass(w.category, DeprecationWarning)
+                and "AsyncClient from pyinjective.async_client is deprecated" in str(w.message)
+            ]
+
+            # Should have exactly one warning
+            assert len(async_client_warnings) == 1
+
+            warning = async_client_warnings[0]
+            # Check warning message contains migration advice
+            assert "Please use AsyncClient from pyinjective.async_client_v2 instead" in str(warning.message)
+            # Check warning category
+            assert warning.category == DeprecationWarning
+            # Check stacklevel is working correctly (should point to this test file)
+            assert "test_async_client_deprecation_warnings.py" in warning.filename
+
+            # Verify the client was still created successfully
+            assert client is not None
+            assert hasattr(client, "network")
+            assert client.network == mock_network

tests/test_indexer_client_deprecation_warnings.py

@@ -0,0 +1,81 @@
+import warnings
+from unittest.mock import MagicMock, patch
+
+from pyinjective.core.network import Network
+
+
+class TestIndexerClientDeprecationWarnings:
+    @patch("pyinjective.indexer_client.IndexerGrpcSpotStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcPortfolioStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcOracleStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcMetaStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcExplorerStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcDerivativeStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcAuctionStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcAccountStream")
+    @patch("pyinjective.indexer_client.IndexerGrpcSpotApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcPortfolioApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcOracleApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcMetaApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcInsuranceApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcExplorerApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcDerivativeApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcAuctionApi")
+    @patch("pyinjective.indexer_client.IndexerGrpcAccountApi")
+    def test_listen_derivative_positions_updates_deprecation_warning(self, *mocks):
+        """Test that calling listen_derivative_positions_updates raises a deprecation warning."""
+        # Create a mock network to avoid actual network initialization
+        mock_network = MagicMock(spec=Network)
+        mock_network.exchange_cookie_assistant = MagicMock()
+        mock_network.explorer_cookie_assistant = MagicMock()
+        mock_network.create_exchange_grpc_channel = MagicMock()
+        mock_network.create_explorer_grpc_channel = MagicMock()
+
+        # Import here to avoid early import issues
+        from pyinjective.indexer_client import IndexerClient
+
+        # Create IndexerClient instance
+        client = IndexerClient(network=mock_network)
+
+        # Mock the derivative_stream_api.stream_positions method to avoid actual streaming
+        client.derivative_stream_api.stream_positions = MagicMock()
+
+        # Capture warnings
+        with warnings.catch_warnings(record=True) as warning_list:
+            warnings.simplefilter("always")  # Ensure all warnings are captured
+
+            # Mock callback function
+            mock_callback = MagicMock()
+
+            # Call the deprecated method - this should trigger the deprecation warning
+            client.listen_derivative_positions_updates(
+                callback=mock_callback, market_ids=["market_1"], subaccount_ids=["subaccount_1"]
+            )
+
+            # Find the deprecation warning
+            deprecation_warnings = [
+                w
+                for w in warning_list
+                if issubclass(w.category, DeprecationWarning)
+                and "This method is deprecated. Use listen_derivative_positions_v2_updates instead" in str(w.message)
+            ]
+
+            # Should have exactly one warning
+            assert len(deprecation_warnings) == 1
+
+            warning = deprecation_warnings[0]
+            # Check warning message contains migration advice
+            assert "Use listen_derivative_positions_v2_updates instead" in str(warning.message)
+            # Check warning category
+            assert warning.category == DeprecationWarning
+            # Check stacklevel is working correctly (should point to this test file)
+            assert "test_indexer_client_deprecation_warnings.py" in warning.filename
+
+            # Verify the underlying method was still called (functionality preserved)
+            client.derivative_stream_api.stream_positions.assert_called_once_with(
+                callback=mock_callback,
+                on_end_callback=None,
+                on_status_callback=None,
+                market_ids=["market_1"],
+                subaccount_ids=["subaccount_1"],
+            )