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)