def test_int_symmetry(self):
"""encode/decode_int should work back and forth."""
result = wire.encode_int(99)
self.assertEqual(wire.decode_int(result), (99, ''))
def test_encode_int_1(self):
self.assertEqual(wire.encode_int(1), '\x01')
self.assertEqual(wire.encode_int(1, 1), '\x01')
def test_encode_int_padded(self):
"""Padding is in the correct place."""
self.assertEqual(wire.encode_int(1, 2), '\x01\x00')
self.assertEqual(wire.encode_int(1, 3), '\x01\x00\x00')
def _encode_int(number, length=1):
return wire.encode_int(number, length)
