test: rewrite test_convert_array_invalid_json using caplog fixture (simpler, robust)

Author: amathxbt

tests/utils_test.py

@@ -1,4 +1,3 @@
-from unittest.mock import patch
 import json
 
 import numpy as np
@@ -69,32 +68,31 @@ def test_convert_array_empty():
     assert result.is_simulation_result is False
 
 
-def test_convert_array_invalid_json():
+def test_convert_array_invalid_json(caplog):
     """Test that invalid JSON tensor data is skipped with a warning instead of crashing.
 
     Previously json.loads() was unguarded and would crash the entire conversion.
     After the fix, malformed JSON tensors are skipped and logged as warnings so
     all other valid tensors in the same response are still returned.
     """
+    import logging
+
     invalid_json_data = [
         [],  # Empty number tensors
         [],  # Empty string tensors
         [["invalid_json", '{"key": invalid}']],  # Invalid JSON
         False,
     ]
 
-    import logging
-    with pytest.warns(None) as warning_list:
-        # Capture logging.warning calls via caplog-style check
-        with patch("opengradient.client._conversions.logging") as mock_log:
-            result = utils.convert_array_to_model_output(invalid_json_data)
+    with caplog.at_level(logging.WARNING):
+        result = utils.convert_array_to_model_output(invalid_json_data)
 
-    # Should not raise; result should have empty jsons dict (bad tensor skipped)
+    # Should not raise; malformed tensor is skipped
     assert isinstance(result, types.ModelOutput)
     assert result.jsons == {}
-    # The warning should have been logged
-    mock_log.warning.assert_called_once()
-    assert "invalid_json" in mock_log.warning.call_args[0][0]
+
+    # A warning should have been emitted mentioning the tensor name
+    assert any("invalid_json" in record.message for record in caplog.records)
 
 
 def test_convert_array_invalid_shape():