def reset(self, ncorrbins=None):
if ncorrbins is not None:
self.ncorrbins = ncorrbins
self.correlograms = CacheMatrix(shape=(0, 0, self.ncorrbins))
self.similarity_matrix = CacheMatrix()
self.similarity_matrix_normalized = None
self.cluster_quality = None
def test_cache_matrix_2():
indices = [2, 3, 5, 7]
matrix = CacheMatrix(shape=(0, 0, 10))
d = {(i, j): i + j for i in indices for j in indices}
matrix.update(indices, d)
assert np.array_equal(matrix.not_in_indices(indices), [])
matrix.invalidate([2, 5])
assert np.array_equal(matrix.not_in_indices(indices), [2, 5])
def test_cache_matrix_2():
indices = [2, 3, 5, 7]
matrix = CacheMatrix(shape=(0, 0, 10))
d = {(i, j): i + j for i in indices for j in indices}
matrix.update(indices, d)
assert np.array_equal(matrix.not_in_indices(indices), [])
matrix.invalidate([2, 5])
assert np.array_equal(matrix.not_in_indices(indices), [2, 5])
def test_cache_matrix_1():
indices = [2, 3, 5, 7]
matrix = CacheMatrix(shape=(0, 0, 10))
assert np.array_equal(matrix.not_in_indices(indices), indices)
d = {(i, j): i + j for i in indices for j in indices}
matrix.update(indices, d)
matrix_actual = (np.array(indices).reshape((-1, 1)) +
np.array(indices).reshape((1, -1)))
assert np.array_equal(matrix.to_array()[:, :, 0], matrix_actual)
assert np.array_equal(matrix.not_in_indices(indices), [])
def test_cache_matrix_1():
indices = [2, 3, 5, 7]
matrix = CacheMatrix(shape=(0, 0, 10))
assert np.array_equal(matrix.not_in_indices(indices), indices)
d = {(i, j): i + j for i in indices for j in indices}
matrix.update(indices, d)
matrix_actual = np.array(indices).reshape((-1, 1)) + np.array(indices).reshape((1, -1))
assert np.array_equal(matrix.to_array()[:, :, 0], matrix_actual)
assert np.array_equal(matrix.not_in_indices(indices), [])
def test_cache_matrix_2():
indices = [2, 3, 5, 7]
matrix = CacheMatrix()
assert np.array_equal(matrix.not_in_key_indices(indices), indices)
matrix.update(2, {(2, 2): 0, (2, 3): 0, (3, 2): 0})
assert np.array_equal(matrix.not_in_key_indices(indices), [3, 5, 7])
matrix.update([2, 3], {(2, 2): 0, (2, 3): 0, (3, 2): 0, (3, 3): 0})
assert np.array_equal(matrix.not_in_key_indices(indices), [5, 7])
matrix.invalidate([2, 5])
assert np.array_equal(matrix.not_in_key_indices(indices), [2, 5, 7])
d = {(i, j): i + j for i in indices for j in indices}
matrix.update(indices, d)
assert np.array_equal(matrix.not_in_key_indices(indices), [])
def test_cache_matrix_2():
indices = [2, 3, 5, 7]
matrix = CacheMatrix()
assert np.array_equal(matrix.not_in_key_indices(indices), indices)
matrix.update(2, {(2, 2): 0, (2, 3): 0, (3, 2): 0})
assert np.array_equal(matrix.not_in_key_indices(indices), [3, 5, 7])
matrix.update([2, 3], {(2, 2): 0, (2, 3): 0, (3, 2): 0, (3, 3): 0})
assert np.array_equal(matrix.not_in_key_indices(indices), [5, 7])
matrix.invalidate([2, 5])
assert np.array_equal(matrix.not_in_key_indices(indices), [2, 5, 7])
d = {(i, j): i + j for i in indices for j in indices}
matrix.update(indices, d)
assert np.array_equal(matrix.not_in_key_indices(indices), [])
