diff options
Diffstat (limited to 'Lib/test/test_socket.py')
| -rw-r--r-- | Lib/test/test_socket.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py index c5975c8..cce7d52 100644 --- a/Lib/test/test_socket.py +++ b/Lib/test/test_socket.py @@ -4629,9 +4629,10 @@ class BufferIOTest(SocketConnectedTest): SocketConnectedTest.__init__(self, methodName=methodName) def testRecvIntoArray(self): - buf = bytearray(1024) + buf = array.array("B", [0] * len(MSG)) nbytes = self.cli_conn.recv_into(buf) self.assertEqual(nbytes, len(MSG)) + buf = buf.tobytes() msg = buf[:len(MSG)] self.assertEqual(msg, MSG) @@ -4658,9 +4659,10 @@ class BufferIOTest(SocketConnectedTest): _testRecvIntoMemoryview = _testRecvIntoArray def testRecvFromIntoArray(self): - buf = bytearray(1024) + buf = array.array("B", [0] * len(MSG)) nbytes, addr = self.cli_conn.recvfrom_into(buf) self.assertEqual(nbytes, len(MSG)) + buf = buf.tobytes() msg = buf[:len(MSG)] self.assertEqual(msg, MSG) |