Fix

Author: jy-tan

drift/instrumentation/psycopg2/instrumentation.py

@@ -843,7 +843,14 @@ def _mock_execute_with_data(self, cursor: Any, mock_data: dict[str, Any]) -> Non
         # If it's a dict cursor and we have description, convert rows to dicts
         if is_dict_cursor and description_data:
             column_names = [col["name"] for col in description_data]
-            mock_rows = [dict(zip(column_names, row)) for row in mock_rows]
+            converted_rows = []
+            for row in mock_rows:
+                if len(column_names) != len(row):
+                    raise ValueError(
+                        f"Column count mismatch: {len(column_names)} columns but row has {len(row)} values"
+                    )
+                converted_rows.append(dict(zip(column_names, row)))
+            mock_rows = converted_rows
 
         cursor._mock_rows = mock_rows  # pyright: ignore[reportAttributeAccessIssue]
         cursor._mock_index = 0  # pyright: ignore[reportAttributeAccessIssue]