def test_getset(self):
indexes = [1, 10, 5, 6, 20, 100, 7, 5]
vm = VM()
byterange.new(vm)
for val in indexes:
vm.push(val * 58)
vm.push(val + 100)
vm.swap2()
byterange.set_val(vm)
for val in indexes:
vm.push(val * 58)
vm.dup1()
byterange.get(vm)
self.assertEqual(val + 100, vm.stack.items[0])
vm.pop()
def test_frombytes(self):
data = bytearray(random.getrandbits(8) for _ in range(500))
data2 = bytearray(data)
if len(data2) % 32 != 0:
data2 = data2 + b"\0" * (32 - (len(data2) % 32))
chunks = [
eth_utils.big_endian_to_int(data2[i:i + 32])
for i in range(0, len(data2), 32)
]
vm = VM()
byterange.new(vm)
for i, chunk in enumerate(chunks):
vm.push(i * 32)
vm.push(chunk)
vm.swap2()
byterange.set_val(vm)
self.assertEqual(byterange.frombytes(data), vm.stack[0])
def test_static_get(self):
vm = VM()
byterange.new(vm)
for val in range(200):
vm.push(val * 58)
vm.push(val + 100)
vm.swap2()
byterange.set_val(vm)
tup = vm.stack.items[0]
for val in range(200):
vm.push(val * 58)
vm.dup1()
byterange.get(vm)
self.assertEqual(byterange.get_static(tup, val * 58),
vm.stack.items[0])
vm.pop()