| 1234567891011121314151617 |
- import numpy as np
- from numpy.testing import assert_equal
- class TestDot:
- 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)
|
