mirror of
https://github.com/bvanroll/college-python-image.git
synced 2025-09-02 14:02:45 +00:00
first commit
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
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()
|
Reference in New Issue
Block a user