def test_linalg_consistent_names():
m, n = 20, 10
mat = np.random.rand(m, n)
data = da.from_array(mat, chunks=(10, n), name="A")
q1, r1 = qr(data)
q2, r2 = qr(data)
assert same_keys(q1, q2)
assert same_keys(r1, r2)
u1, s1, v1 = svd(data)
u2, s2, v2 = svd(data)
assert same_keys(u1, u2)
assert same_keys(s1, s2)
assert same_keys(v1, v2)
def test_linalg_consistent_names():
m, n = 20, 10
mat = np.random.rand(m, n)
data = da.from_array(mat, chunks=(10, n), name='A')
q1, r1 = qr(data)
q2, r2 = qr(data)
assert same_keys(q1, q2)
assert same_keys(r1, r2)
u1, s1, v1 = svd(data)
u2, s2, v2 = svd(data)
assert same_keys(u1, u2)
assert same_keys(s1, s2)
assert same_keys(v1, v2)
def test_qr(m, n, chunks, error_type):
mat = np.random.rand(m, n)
data = da.from_array(mat, chunks=chunks, name="A")
m_q = m
n_q = min(m, n)
m_r = n_q
n_r = n
m_qtq = n_q
if error_type is None:
q, r = qr(data)
assert_eq((m_q, n_q), q.shape) # shape check
assert_eq((m_r, n_r), r.shape) # shape check
assert_eq(mat, da.dot(q, r)) # accuracy check
assert_eq(np.eye(m_qtq, m_qtq), da.dot(q.T, q)) # q must be orthonormal
assert_eq(r, da.triu(r.rechunk(r.shape[0]))) # r must be upper triangular
else:
with pytest.raises(error_type):
q, r = qr(data)
def test_qr(m, n, chunks, error_type):
mat = np.random.rand(m, n)
data = da.from_array(mat, chunks=chunks, name='A')
m_q = m
n_q = min(m, n)
m_r = n_q
n_r = n
m_qtq = n_q
if error_type is None:
q, r = qr(data)
assert_eq((m_q, n_q), q.shape) # shape check
assert_eq((m_r, n_r), r.shape) # shape check
assert_eq(mat, da.dot(q, r)) # accuracy check
assert_eq(np.eye(m_qtq, m_qtq), da.dot(q.T, q)) # q must be orthonormal
assert_eq(r, da.triu(r.rechunk(r.shape[0]))) # r must be upper triangular
else:
with pytest.raises(error_type):
q, r = qr(data)
