def test_sizeof_with_prepends(self):
r = KeyspacesRegion(
[(Signal(BitField(32)), {})],
fields=[],
prepend_num_keyspaces=True
)
assert r.sizeof(slice(None)) == 4
def test_sizeof_partitioned(self):
r = KeyspacesRegion([(Signal(BitField(32)), {})] * 4,
fields=[mock.Mock()],
partitioned_by_atom=True,
prepend_num_keyspaces=False)
assert r.sizeof(slice(1, 2)) == 4
assert r.sizeof(slice(2, 4)) == 8
def test_sizeof_partitioned(self):
r = KeyspacesRegion(
[(Signal(BitField(32)), {})]*4,
fields=[mock.Mock()],
partitioned_by_atom=True,
prepend_num_keyspaces=False
)
assert r.sizeof(slice(1, 2)) == 4
assert r.sizeof(slice(2, 4)) == 8
def test_sizeof_no_prepends(self, key_bits, n_keys, n_fields, partitioned,
vertex_slice):
# Generate the list of keys, prepends and fields
keys = [BitField(key_bits) for _ in range(n_keys)]
fields = [mock.Mock() for _ in range(n_fields)]
# Create the region
r = KeyspacesRegion(keys, fields, partitioned)
# Determine the size
n_atoms = (n_keys if not partitioned else
vertex_slice.stop - vertex_slice.start)
assert r.sizeof(vertex_slice) == n_atoms * n_fields * 4
def test_sizeof_no_prepends(self, key_bits, n_keys, n_fields, partitioned,
vertex_slice):
# Generate the list of keys, prepends and fields
keys = [(Signal(BitField(key_bits)), {}) for _ in range(n_keys)]
fields = [mock.Mock() for _ in range(n_fields)]
# Create the region
r = KeyspacesRegion(keys, fields, partitioned)
# Determine the size
n_atoms = (n_keys if not partitioned else vertex_slice.stop -
vertex_slice.start)
assert r.sizeof(vertex_slice) == n_atoms * n_fields * 4
def test_sizeof_with_prepends(self):
r = KeyspacesRegion([(Signal(BitField(32)), {})],
fields=[],
prepend_num_keyspaces=True)
assert r.sizeof(slice(None)) == 4
