diff DEPENDENCIES/mingw32/Python27/Lib/site-packages/numpy/tests/test_matlib.py @ 87:2a2c65a20a8b

Add Python libs and headers
author Chris Cannam
date Wed, 25 Feb 2015 14:05:22 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/DEPENDENCIES/mingw32/Python27/Lib/site-packages/numpy/tests/test_matlib.py	Wed Feb 25 14:05:22 2015 +0000
@@ -0,0 +1,55 @@
+from __future__ import division, absolute_import, print_function
+
+import numpy as np
+import numpy.matlib
+from numpy.testing import assert_array_equal, assert_, run_module_suite
+
+def test_empty():
+    x = np.matlib.empty((2,))
+    assert_(isinstance(x, np.matrix))
+    assert_(x.shape, (1, 2))
+
+def test_ones():
+    assert_array_equal(np.matlib.ones((2, 3)),
+                       np.matrix([[ 1.,  1.,  1.],
+                                 [ 1.,  1.,  1.]]))
+
+    assert_array_equal(np.matlib.ones(2), np.matrix([[ 1.,  1.]]))
+
+def test_zeros():
+    assert_array_equal(np.matlib.zeros((2, 3)),
+                       np.matrix([[ 0.,  0.,  0.],
+                                 [ 0.,  0.,  0.]]))
+
+    assert_array_equal(np.matlib.zeros(2), np.matrix([[ 0.,  0.]]))
+
+def test_identity():
+    x = np.matlib.identity(2, dtype=np.int)
+    assert_array_equal(x, np.matrix([[1, 0], [0, 1]]))
+
+def test_eye():
+    x = np.matlib.eye(3, k=1, dtype=int)
+    assert_array_equal(x, np.matrix([[ 0,  1,  0],
+                                     [ 0,  0,  1],
+                                     [ 0,  0,  0]]))
+
+def test_rand():
+    x = np.matlib.rand(3)
+    # check matrix type, array would have shape (3,)
+    assert_(x.ndim == 2)
+
+def test_randn():
+    x = np.matlib.randn(3)
+    # check matrix type, array would have shape (3,)
+    assert_(x.ndim == 2)
+
+def test_repmat():
+    a1 = np.arange(4)
+    x = np.matlib.repmat(a1, 2, 2)
+    y = np.array([[0, 1, 2, 3, 0, 1, 2, 3],
+                  [0, 1, 2, 3, 0, 1, 2, 3]])
+    assert_array_equal(x, y)
+
+
+if __name__ == "__main__":
+    run_module_suite()