94 lines
2.0 KiB
Python
94 lines
2.0 KiB
Python
""" Test functions for linalg module using the matrix class."""
|
|
import numpy as np
|
|
|
|
from numpy.linalg.tests.test_linalg import (
|
|
LinalgCase, apply_tag, TestQR as _TestQR, LinalgTestCase,
|
|
_TestNorm2D, _TestNormDoubleBase, _TestNormSingleBase, _TestNormInt64Base,
|
|
SolveCases, InvCases, EigvalsCases, EigCases, SVDCases, CondCases,
|
|
PinvCases, DetCases, LstsqCases)
|
|
|
|
|
|
CASES = []
|
|
|
|
# square test cases
|
|
CASES += apply_tag('square', [
|
|
LinalgCase("0x0_matrix",
|
|
np.empty((0, 0), dtype=np.double).view(np.matrix),
|
|
np.empty((0, 1), dtype=np.double).view(np.matrix),
|
|
tags={'size-0'}),
|
|
LinalgCase("matrix_b_only",
|
|
np.array([[1., 2.], [3., 4.]]),
|
|
np.matrix([2., 1.]).T),
|
|
LinalgCase("matrix_a_and_b",
|
|
np.matrix([[1., 2.], [3., 4.]]),
|
|
np.matrix([2., 1.]).T),
|
|
])
|
|
|
|
# hermitian test-cases
|
|
CASES += apply_tag('hermitian', [
|
|
LinalgCase("hmatrix_a_and_b",
|
|
np.matrix([[1., 2.], [2., 1.]]),
|
|
None),
|
|
])
|
|
# No need to make generalized or strided cases for matrices.
|
|
|
|
|
|
class MatrixTestCase(LinalgTestCase):
|
|
TEST_CASES = CASES
|
|
|
|
|
|
class TestSolveMatrix(SolveCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestInvMatrix(InvCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestEigvalsMatrix(EigvalsCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestEigMatrix(EigCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestSVDMatrix(SVDCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestCondMatrix(CondCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestPinvMatrix(PinvCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestDetMatrix(DetCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class TestLstsqMatrix(LstsqCases, MatrixTestCase):
|
|
pass
|
|
|
|
|
|
class _TestNorm2DMatrix(_TestNorm2D):
|
|
array = np.matrix
|
|
|
|
|
|
class TestNormDoubleMatrix(_TestNorm2DMatrix, _TestNormDoubleBase):
|
|
pass
|
|
|
|
|
|
class TestNormSingleMatrix(_TestNorm2DMatrix, _TestNormSingleBase):
|
|
pass
|
|
|
|
|
|
class TestNormInt64Matrix(_TestNorm2DMatrix, _TestNormInt64Base):
|
|
pass
|
|
|
|
|
|
class TestQRMatrix(_TestQR):
|
|
array = np.matrix
|