관리-도구

편집 파일: test_numeric.py

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()