Fold the new checks into the existing partial-write tests

Address review: test_write_ready_partial and test_write_sendmsg_partial now run for both bytes and memoryview input and check that the remainder is a view on the original object. Wrap the NEWS entry at 78 columns.

Author: Ar-maan05

Lib/test/test_asyncio/test_selector_events.py

@@ -958,35 +958,22 @@ def test_write_sendmsg_full(self):
 
     @unittest.skipUnless(selector_events._HAS_SENDMSG, 'no sendmsg')
     def test_write_sendmsg_partial(self):
-
-        data = memoryview(b'data')
-        self.sock.sendmsg = mock.Mock()
-        # Sent partial data
-        self.sock.sendmsg.return_value = 2
-
-        transport = self.socket_transport(sendmsg=True)
-        transport._buffer.append(data)
-        self.loop._add_writer(7, transport._write_ready)
-        transport._write_ready()
-        self.assertTrue(self.sock.sendmsg.called)
-        self.assertTrue(self.loop.writers)
-        self.assertEqual(list_to_buffer([b'ta']), transport._buffer)
-
-    @unittest.skipUnless(selector_events._HAS_SENDMSG, 'no sendmsg')
-    def test_write_sendmsg_partial_bytes_no_copy(self):
-        # gh-159503: the unsent remainder of a bytes object must not be
-        # copied, otherwise a large write costs O(n**2).
         data = b'data'
-        self.sock.sendmsg = mock.Mock()
-        self.sock.sendmsg.return_value = 2
-
-        transport = self.socket_transport(sendmsg=True)
-        transport._buffer.append(data)
-        self.loop._add_writer(7, transport._write_ready)
-        transport._write_ready()
-        self.assertEqual(list_to_buffer([b'ta']), transport._buffer)
-        self.assertIsInstance(transport._buffer[0], memoryview)
-        self.assertIs(transport._buffer[0].obj, data)
+        for buf in data, memoryview(data):
+            with self.subTest(type=type(buf).__name__):
+                self.sock.sendmsg = mock.Mock()
+                # Sent partial data
+                self.sock.sendmsg.return_value = 2
+
+                transport = self.socket_transport(sendmsg=True)
+                transport._buffer.append(buf)
+                self.loop._add_writer(7, transport._write_ready)
+                transport._write_ready()
+                self.assertTrue(self.sock.sendmsg.called)
+                self.assertTrue(self.loop.writers)
+                self.assertEqual(list_to_buffer([b'ta']), transport._buffer)
+                # gh-159503: the remainder is a view, not a copy
+                self.assertIs(transport._buffer[0].obj, data)
 
     @unittest.skipUnless(selector_events._HAS_SENDMSG, 'no sendmsg')
     def test_write_sendmsg_half_buffer(self):
@@ -1087,29 +1074,19 @@ def test_write_ready_no_data(self):
         self.assertRaises(AssertionError, transport._write_ready)
 
     def test_write_ready_partial(self):
-        data = memoryview(b'data')
-        self.sock.send.return_value = 2
-
-        transport = self.socket_transport()
-        transport._buffer.append(data)
-        self.loop._add_writer(7, transport._write_ready)
-        transport._write_ready()
-        self.loop.assert_writer(7, transport._write_ready)
-        self.assertEqual(list_to_buffer([b'ta']), transport._buffer)
-
-    def test_write_ready_partial_bytes_no_copy(self):
-        # gh-159503: the unsent remainder of a bytes object must not be
-        # copied, otherwise a large write costs O(n**2).
         data = b'data'
-        self.sock.send.return_value = 2
-
-        transport = self.socket_transport()
-        transport._buffer.append(data)
-        self.loop._add_writer(7, transport._write_ready)
-        transport._write_ready()
-        self.assertEqual(list_to_buffer([b'ta']), transport._buffer)
-        self.assertIsInstance(transport._buffer[0], memoryview)
-        self.assertIs(transport._buffer[0].obj, data)
+        for buf in data, memoryview(data):
+            with self.subTest(type=type(buf).__name__):
+                self.sock.send.return_value = 2
+
+                transport = self.socket_transport()
+                transport._buffer.append(buf)
+                self.loop._add_writer(7, transport._write_ready)
+                transport._write_ready()
+                self.loop.assert_writer(7, transport._write_ready)
+                self.assertEqual(list_to_buffer([b'ta']), transport._buffer)
+                # gh-159503: the remainder is a view, not a copy
+                self.assertIs(transport._buffer[0].obj, data)
 
     def test_write_ready_partial_none(self):
         data = b'data'

Misc/NEWS.d/next/Library/2026-10-16-01-56-15.gh-issue-159503.WgYiow.rst

@@ -1 +1,4 @@
-Fix quadratic copying in :mod:`asyncio` selector socket transports when a large :class:`bytes` write is only partly sent while other data is buffered. The unsent remainder is now kept as a :class:`memoryview` instead of being copied on every partial send.
+Fix quadratic copying in :mod:`asyncio` selector socket transports when a
+large :class:`bytes` write is only partly sent while other data is buffered.
+The unsent remainder is now kept as a :class:`memoryview` instead of being
+copied on every partial send.