buf.putu8(0x00)
putfn(buf, T.bytes_as_int(w, bigendp))
buf.putu8(w + 1)
- me.assertEqual(C.ByteString(buf), T.span(w + 2))
- me.assertEqual(C.ByteString(putfn(C.WriteBuffer(), (1 << 8*w) - 1)),
+ me.assertEqual(buf.contents, T.span(w + 2))
+ me.assertEqual(putfn(C.WriteBuffer(), (1 << 8*w) - 1).contents,
w*C.bytes("ff"))
- me.assertEqual(C.ByteString(putfn(C.WriteBuffer(), C.MP(0))),
- w*C.bytes("00"))
+ me.assertEqual(putfn(C.WriteBuffer(), C.MP(0)).contents, w*C.bytes("00"))
## Check overflow detection.
- me.assertRaises((OverflowError, ValueError),
- putfn, C.WriteBuffer(), 1 << 8*w)
+ me.assertRaises(OverflowError, putfn, C.WriteBuffer(), 1 << 8*w)
def check_putbufn(me, w, bigendp, putfn):
"""Check `putblkN'."""
## Go through a number of different sizes.
for n in [0, 1, 7, 8, 19, 255, 12345, 65535, 123456]:
if n >= 1 << 8*w: continue
- me.assertEqual(C.ByteString(putfn(C.WriteBuffer().putu8(0x00),
- T.span(n)).putu8(0xff)),
+ me.assertEqual(putfn(C.WriteBuffer().putu8(0x00),
+ T.span(n)).putu8(0xff).contents,
T.prep_lenseq(w, n, bigendp, True))
## Check blocks which are too large for the length prefix.
buf.zero(17)
buf.put(T.span(23))
me.assertEqual(buf.size, 40)
- me.assertEqual(C.ByteString(buf), C.ByteString.zero(17) + T.span(23))
+ me.assertEqual(buf, C.ByteString.zero(17) + T.span(23))
+ me.assertEqual(buf.contents, C.ByteString.zero(17) + T.span(23))
###----- That's all, folks --------------------------------------------------