24 lines
585 B
Python
24 lines
585 B
Python
|
from __future__ import division, absolute_import, print_function
|
||
|
|
||
|
import numpy as np
|
||
|
from numpy.testing import assert_equal, run_module_suite
|
||
|
|
||
|
class TestDot(object):
|
||
|
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()
|