o 6a@s.ddlZddlmZmZmZGdddZdS)N)assert_ assert_equal assert_raisesc@s,eZdZddZddZddZddZd S) TestRegressioncCs(td}ttt||t|dS)Nz [1 0; 1 0])npmatrixrtypekronselfxr G/usr/lib/python3/dist-packages/numpy/matrixlib/tests/test_regression.pytest_kron_matrixs zTestRegression.test_kron_matrixcCsttjdgtd}tt|jtjutt|jtjutdg\}}tt|tjutt|tjudS)Ng?)dtypeg) rrfloatrrrealimagnonzerondarray)r acdr r rtest_matrix_properties s z%TestRegression.test_matrix_propertiescCsdd}tt|dS)NcSsttdtddS)N)rmateyeonesr r r rmulsz=TestRegression.test_matrix_multiply_by_1d_vector..mul)r ValueError)r rr r r!test_matrix_multiply_by_1d_vectorsz0TestRegression.test_matrix_multiply_by_1d_vectorcCs:ttjddd}t|jdt|jddS)Nr)r"r )rasmatrixrandomuniformrstdshapeargmaxr r r rtest_matrix_std_argmaxsz%TestRegression.test_matrix_std_argmaxN)__name__ __module__ __qualname__rrr r)r r r rrs  r)numpyr numpy.testingrrrrr r r rs