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_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), [])
