def test_kronsum(self):
cases = []
cases.append(array([[0]]))
cases.append(array([[-1]]))
cases.append(array([[4]]))
cases.append(array([[10]]))
cases.append(array([[1, 2], [3, 4]]))
cases.append(array([[0, 2], [5, 0]]))
cases.append(array([[0, 2, -6], [8, 0, 14], [0, 3, 0]]))
cases.append(array([[1, 0, 0], [0, 5, -1], [4, -2, 8]]))
for a in cases:
for b in cases:
result = construct.kronsum(csr_matrix(a), csr_matrix(b)).todense()
expected = np.kron(np.eye(len(b)), a) + np.kron(b, np.eye(len(a)))
assert_array_equal(result, expected)
def test_kronsum(self):
cases = []
cases.append(array([[0]]))
cases.append(array([[-1]]))
cases.append(array([[4]]))
cases.append(array([[10]]))
cases.append(array([[1,2],[3,4]]))
cases.append(array([[0,2],[5,0]]))
cases.append(array([[0,2,-6],[8,0,14],[0,3,0]]))
cases.append(array([[1,0,0],[0,5,-1],[4,-2,8]]))
for a in cases:
for b in cases:
result = construct.kronsum(csr_matrix(a),csr_matrix(b)).todense()
expected = np.kron(np.eye(len(b)), a) + \
np.kron(b, np.eye(len(a)))
assert_array_equal(result,expected)
