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)