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