1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
| 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)
|
|
