Python UmfpackSolver Examples

Example #1

File: test_umfpack_get_LU_cysparse_INT32_t_COMPLEX128_t_CSRSparseMatrix.py Project: fagan2888/SuiteSparse.py

    def setUp(self):

        self.A = LinearFillLLSparseMatrix(size=SIZE,
                                          dtype=COMPLEX128_T,
                                          itype=INT32_T)

        self.C = self.A.to_csr()

        self.solver = UmfpackSolver(self.C)

Example #2

File: test_umfpack_get_LU_cysparse_INT32_t_FLOAT64_t_LLSparseMatrix.py Project: fagan2888/SuiteSparse.py

    def setUp(self):

        self.A = LinearFillLLSparseMatrix(size=SIZE,
                                          dtype=FLOAT64_T,
                                          itype=INT32_T)

        self.C = self.A

        self.solver = UmfpackSolver(self.C)

Example #3

File: test_umfpack_get_LU_cysparse_INT32_t_FLOAT64_t_CSCSparseMatrix.py Project: fagan2888/SuiteSparse.py

    def setUp(self):

        self.A = LinearFillLLSparseMatrix(size=SIZE, dtype=FLOAT64_T, itype=INT32_T, store_zero=True)


        self.C = self.A.to_csc()



        self.solver = UmfpackSolver(self.C)

Example #4

    def setUp(self):

        self.A = LinearFillLLSparseMatrix(size=SIZE,
                                          dtype=COMPLEX128_T,
                                          itype=INT64_T,
                                          store_zero=True)

        self.C = self.A

        self.solver = UmfpackSolver(self.C)

Example #5

File: test_umfpack_get_LU_cysparse_INT32_t_COMPLEX128_t_CSRSparseMatrix.py Project: fagan2888/SuiteSparse.py

    def setUp(self):

        self.size = SIZE

        self.A = LinearFillLLSparseMatrix(size=self.size,
                                          dtype=COMPLEX128_T,
                                          itype=INT32_T,
                                          store_symmetry=True)

        self.C = self.A.to_csr()

        self.solver = UmfpackSolver(self.C)

Example #6

File: test_umfpack_get_LU_cysparse_INT32_t_FLOAT64_t_LLSparseMatrix.py Project: fagan2888/SuiteSparse.py

    def setUp(self):

        self.size = SIZE

        self.A = LinearFillLLSparseMatrix(size=self.size,
                                          dtype=FLOAT64_T,
                                          itype=INT32_T,
                                          store_symmetry=True)

        self.C = self.A

        self.solver = UmfpackSolver(self.C)

Example #7

File: test_umfpack_get_LU_cysparse_INT32_t_COMPLEX128_t_CSRSparseMatrix.py Project: PythonOptimizers/SuiteSparse.py

    def setUp(self):

        self.A = LinearFillLLSparseMatrix(size=SIZE, dtype=COMPLEX128_T, itype=INT32_T)

        self.C = self.A.to_csr()

        self.solver = UmfpackSolver(self.C)

Example #8

File: test_umfpack_get_LU_cysparse_INT32_t_COMPLEX128_t_CSRSparseMatrix.py Project: fagan2888/SuiteSparse.py

class CySparseUmfpackget_LUWithSymmetrySWithZero_CSRSparseMatrix_INT32_t_COMPLEX128_t_TestCase(
        unittest.TestCase):
    def setUp(self):

        self.size = SIZE

        self.A = LinearFillLLSparseMatrix(size=self.size,
                                          dtype=COMPLEX128_T,
                                          itype=INT32_T,
                                          store_symmetry=True,
                                          store_zero=True)

        self.C = self.A.to_csr()

        self.solver = UmfpackSolver(self.C)

    def test_factorization_element_by_element(self):
        """
        Verify the equality ``L * U = P * R * A * Q``
        """
        (L, U, P, Q, D, do_recip, R) = self.solver.get_LU()

        lhs = L * U

        P_mat = PermutationLLSparseMatrix(P=P,
                                          size=SIZE,
                                          dtype=COMPLEX128_T,
                                          itype=INT32_T)
        Q_mat = PermutationLLSparseMatrix(P=Q,
                                          size=SIZE,
                                          dtype=COMPLEX128_T,
                                          itype=INT32_T)

        R_mat = None
        if do_recip:
            R_mat = LLSparseMatrix(size=SIZE,
                                   dtype=COMPLEX128_T,
                                   itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = R[i]
        else:
            R_mat = LLSparseMatrix(size=size,
                                   dtype=COMPLEX128_T,
                                   itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = 1 / R[i]

        rhs = P_mat * R_mat * self.C * Q_mat

        for i in xrange(SIZE):
            for j in xrange(SIZE):

                self.assertTrue(
                    abs(lhs[i, j] - rhs[i, j]) < EPS,
                    "lhs[%d, %d] =? %f + %f j , rhs[%d, %d] = %f + %f j" %
                    (i, j, lhs[i, j].real, lhs[i, j].imag, i, j,
                     rhs[i, j].real, rhs[i, j].imag))

Example #9

File: test_umfpack_get_LU_cysparse_INT32_t_COMPLEX128_t_CSRSparseMatrix.py Project: PythonOptimizers/SuiteSparse.py

    def setUp(self):

        self.size = SIZE

        self.A = LinearFillLLSparseMatrix(size=self.size, dtype=COMPLEX128_T, itype=INT32_T, store_symmetry=True)

        self.C = self.A.to_csr()

        self.solver = UmfpackSolver(self.C)

Example #10

File: test_umfpack_get_LU_cysparse_INT32_t_COMPLEX128_t_LLSparseMatrix.py Project: PythonOptimizers/SuiteSparse.py

    def setUp(self):

        self.A = LinearFillLLSparseMatrix(size=SIZE, dtype=COMPLEX128_T, itype=INT32_T, store_zero=True)


        self.C = self.A



        self.solver = UmfpackSolver(self.C)

Example #11

File: test_umfpack_get_LU_cysparse_INT64_t_FLOAT64_t_LLSparseMatrix.py Project: PythonOptimizers/SuiteSparse.py

    def setUp(self):
       

        self.A = LinearFillLLSparseMatrix(size=SIZE, dtype=FLOAT64_T, itype=INT64_T)


        self.C = self.A



        self.solver = UmfpackSolver(self.C)

Example #12

File: test_umfpack_get_LU_cysparse_INT64_t_FLOAT64_t_LLSparseMatrix.py Project: PythonOptimizers/SuiteSparse.py

    def setUp(self):

        self.size = SIZE

        self.A = LinearFillLLSparseMatrix(size=self.size, dtype=FLOAT64_T, itype=INT64_T, store_symmetry=True)


        self.C = self.A



        self.solver = UmfpackSolver(self.C)

Example #13

File: test_umfpack_get_LU_cysparse_INT32_t_FLOAT64_t_LLSparseMatrix.py Project: fagan2888/SuiteSparse.py

class CySparseUmfpackget_LUNoSymmetryNoZero_LLSparseMatrix_INT32_t_FLOAT64_t_TestCase(
        unittest.TestCase):
    def setUp(self):

        self.A = LinearFillLLSparseMatrix(size=SIZE,
                                          dtype=FLOAT64_T,
                                          itype=INT32_T)

        self.C = self.A

        self.solver = UmfpackSolver(self.C)

    def test_factorization_element_by_element(self):
        """
        Verify the equality ``L * U = P * R * A * Q``
        """
        (L, U, P, Q, D, do_recip, R) = self.solver.get_LU()

        lhs = L * U

        P_mat = PermutationLLSparseMatrix(P=P,
                                          size=SIZE,
                                          dtype=FLOAT64_T,
                                          itype=INT32_T)
        Q_mat = PermutationLLSparseMatrix(P=Q,
                                          size=SIZE,
                                          dtype=FLOAT64_T,
                                          itype=INT32_T)

        R_mat = None
        if do_recip:
            R_mat = LLSparseMatrix(size=SIZE, dtype=FLOAT64_T, itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = R[i]
        else:
            R_mat = LLSparseMatrix(size=size, dtype=FLOAT64_T, itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = 1 / R[i]

        rhs = P_mat * R_mat * self.C * Q_mat

        for i in xrange(SIZE):
            for j in xrange(SIZE):

                self.assertTrue(
                    lhs[i, j] - rhs[i, j] < EPS,
                    "lhs[%d, %d] =? %f , rhs[%d, %d] = %f" %
                    (i, j, lhs[i, j], i, j, rhs[i, j]))

Example #14

File: test_umfpack_get_LU_cysparse_INT32_t_COMPLEX128_t_CSRSparseMatrix.py Project: PythonOptimizers/SuiteSparse.py

class CySparseUmfpackget_LUWithSymmetrySWithZero_CSRSparseMatrix_INT32_t_COMPLEX128_t_TestCase(unittest.TestCase):
    def setUp(self):

        self.size = SIZE

        self.A = LinearFillLLSparseMatrix(
            size=self.size, dtype=COMPLEX128_T, itype=INT32_T, store_symmetry=True, store_zero=True
        )

        self.C = self.A.to_csr()

        self.solver = UmfpackSolver(self.C)

    def test_factorization_element_by_element(self):
        """
        Verify the equality ``L * U = P * R * A * Q``
        """
        (L, U, P, Q, D, do_recip, R) = self.solver.get_LU()

        lhs = L * U

        P_mat = PermutationLLSparseMatrix(P=P, size=SIZE, dtype=COMPLEX128_T, itype=INT32_T)
        Q_mat = PermutationLLSparseMatrix(P=Q, size=SIZE, dtype=COMPLEX128_T, itype=INT32_T)

        R_mat = None
        if do_recip:
            R_mat = LLSparseMatrix(size=SIZE, dtype=COMPLEX128_T, itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = R[i]
        else:
            R_mat = LLSparseMatrix(size=size, dtype=COMPLEX128_T, itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = 1 / R[i]

        rhs = P_mat * R_mat * self.C * Q_mat

        for i in xrange(SIZE):
            for j in xrange(SIZE):

                self.assertTrue(
                    abs(lhs[i, j] - rhs[i, j]) < EPS,
                    "lhs[%d, %d] =? %f + %f j , rhs[%d, %d] = %f + %f j"
                    % (i, j, lhs[i, j].real, lhs[i, j].imag, i, j, rhs[i, j].real, rhs[i, j].imag),
                )

Example #15

File: test_umfpack_get_LU_cysparse_INT32_t_FLOAT64_t_CSCSparseMatrix.py Project: PythonOptimizers/SuiteSparse.py

class CySparseUmfpackget_LUNoSymmetryNoZero_CSCSparseMatrix_INT32_t_FLOAT64_t_TestCase(unittest.TestCase):
    def setUp(self):
       

        self.A = LinearFillLLSparseMatrix(size=SIZE, dtype=FLOAT64_T, itype=INT32_T)


        self.C = self.A.to_csc()



        self.solver = UmfpackSolver(self.C)

    def test_factorization_element_by_element(self):
        """
        Verify the equality ``L * U = P * R * A * Q``
        """
        (L, U, P, Q, D, do_recip, R) = self.solver.get_LU()

        lhs = L * U

        P_mat = PermutationLLSparseMatrix(P=P, size=SIZE, dtype=FLOAT64_T, itype=INT32_T)
        Q_mat = PermutationLLSparseMatrix(P=Q, size=SIZE, dtype=FLOAT64_T, itype=INT32_T)

        R_mat = None
        if do_recip:
            R_mat = LLSparseMatrix(size=SIZE, dtype=FLOAT64_T, itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = R[i]
        else:
            R_mat = LLSparseMatrix(size=size, dtype=FLOAT64_T, itype=INT32_T)
            for i in xrange(SIZE):
                R_mat[i, i] = 1/R[i]

        rhs = P_mat * R_mat * self.C * Q_mat

        for i in xrange(SIZE):
            for j in xrange(SIZE):

                self.assertTrue(lhs[i, j] - rhs[i, j] < EPS, "lhs[%d, %d] =? %f , rhs[%d, %d] = %f" % (i,j, lhs[i, j], i, j, rhs[i, j]))