Server : Apache System : Linux indy02.toastserver.com 3.10.0-962.3.2.lve1.5.85.el7.x86_64 #1 SMP Thu Apr 18 15:18:36 UTC 2024 x86_64 User : palandch ( 1163) PHP Version : 7.1.33 Disable Function : NONE Directory :  /opt/alt/python27/lib64/python2.7/site-packages/numpy/matrixlib/tests/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

from __future__ import division, absolute_import, print_function

import numpy as np
from numpy.testing import assert_equal, TestCase, run_module_suite

class TestDot(TestCase):
    def test_matscalar(self):
        b1 = np.matrix(np.ones((3, 3), dtype=complex))
        assert_equal(b1*1.0, b1)


def test_diagonal():
    b1 = np.matrix([[1,2],[3,4]])
    diag_b1 = np.matrix([[1, 4]])
    array_b1 = np.array([1, 4])

    assert_equal(b1.diagonal(), diag_b1)
    assert_equal(np.diagonal(b1), array_b1)
    assert_equal(np.diag(b1), array_b1)


if __name__ == "__main__":
    run_module_suite()