def test_new_empty_writer(self):
writer = LittleEndianByteArrayByteWriter()
self.assertEqual(writer.get_n_bytes_written(),0)
self.assertEqual(writer.data,bytearray())
def test_write_long(self):
writer = LittleEndianByteArrayByteWriter()
long_to_write = 0xF1F2F3F4F5F6F7F8
writer.write_long(long_to_write)
self.assertEqual(writer.get_n_bytes_written(),8)
self.assertEqual(writer.data,bytearray([0xF8,0xF7, 0xF6, 0xF5, 0xF4, 0xF3, 0xF2,0xF1]))
def test_write_int(self):
writer = LittleEndianByteArrayByteWriter()
int_to_write = 0xF1F2F3F4
writer.write_int(int_to_write)
self.assertEqual(writer.get_n_bytes_written(),4)
self.assertEqual(writer.data,bytearray([0xF4, 0xF3, 0xF2,0xF1]))
def test_write_short(self):
writer = LittleEndianByteArrayByteWriter()
short_to_write = 0xF1F2
writer.write_short(short_to_write)
self.assertEqual(writer.get_n_bytes_written(),2)
self.assertEqual(writer.data,bytearray([0xF2,0xF1]))
def test_write_bytes(self):
writer = LittleEndianByteArrayByteWriter()
bytes_to_write = [1,2,3,4,5,6]
writer.write_bytes(bytes_to_write)
self.assertEqual(writer.get_n_bytes_written(),len(bytes_to_write))
self.assertEqual(writer.data,bytearray(bytes_to_write))
def test_write_byte_truncate(self):
writer = LittleEndianByteArrayByteWriter()
writer.write_byte(0xFF37)
self.assertEqual(writer.get_n_bytes_written(),1)
self.assertEqual(writer.data[0],0x37)
