testUSB1: Add tests for USBTransfer bytearray buffer.
diff --git a/usb1/testUSB1.py b/usb1/testUSB1.py
index 180f70d..d213239 100644
--- a/usb1/testUSB1.py
+++ b/usb1/testUSB1.py
@@ -23,13 +23,15 @@
import libusb1
from ctypes import pointer
+buff_len = 1024
+buffer_base = [x % 256 for x in xrange(buff_len)]
if sys.version_info[0] == 3:
- buff = bytes([0, 0xff])
- other_buff = bytes((ord(x) for x in 'foo'))
+ buff = bytes(buffer_base)
+ other_buff = bytes(reversed(buffer_base))
else:
- buff = '\x00\xff'
- other_buff = 'foo'
-buff_len = 2
+ buff = ''.join(chr(x) for x in buffer_base)
+ other_buff = ''.join(chr(x) for x in reversed(buffer_base))
+bytearray_buff = bytearray(buffer_base)
class USBContext(usb1.USBContext):
def open(self):
@@ -129,6 +131,8 @@
self.assertEqual(buff, transfer.getBuffer())
transfer.setBuffer(other_buff)
self.assertEqual(other_buff, transfer.getBuffer())
+ transfer.setBuffer(bytearray_buff)
+ self.assertEqual(bytearray_buff, transfer.getBuffer())
transfer.setBuffer(buff_len)
self.assertEqual(buff_len, len(transfer.getBuffer()))
# All provided, buffer length variant
