Chris@87: from __future__ import division, absolute_import, print_function
Chris@87: 
Chris@87: import numpy as np
Chris@87: import numpy.matlib
Chris@87: from numpy.testing import assert_array_equal, assert_, run_module_suite
Chris@87: 
Chris@87: def test_empty():
Chris@87:     x = np.matlib.empty((2,))
Chris@87:     assert_(isinstance(x, np.matrix))
Chris@87:     assert_(x.shape, (1, 2))
Chris@87: 
Chris@87: def test_ones():
Chris@87:     assert_array_equal(np.matlib.ones((2, 3)),
Chris@87:                        np.matrix([[ 1.,  1.,  1.],
Chris@87:                                  [ 1.,  1.,  1.]]))
Chris@87: 
Chris@87:     assert_array_equal(np.matlib.ones(2), np.matrix([[ 1.,  1.]]))
Chris@87: 
Chris@87: def test_zeros():
Chris@87:     assert_array_equal(np.matlib.zeros((2, 3)),
Chris@87:                        np.matrix([[ 0.,  0.,  0.],
Chris@87:                                  [ 0.,  0.,  0.]]))
Chris@87: 
Chris@87:     assert_array_equal(np.matlib.zeros(2), np.matrix([[ 0.,  0.]]))
Chris@87: 
Chris@87: def test_identity():
Chris@87:     x = np.matlib.identity(2, dtype=np.int)
Chris@87:     assert_array_equal(x, np.matrix([[1, 0], [0, 1]]))
Chris@87: 
Chris@87: def test_eye():
Chris@87:     x = np.matlib.eye(3, k=1, dtype=int)
Chris@87:     assert_array_equal(x, np.matrix([[ 0,  1,  0],
Chris@87:                                      [ 0,  0,  1],
Chris@87:                                      [ 0,  0,  0]]))
Chris@87: 
Chris@87: def test_rand():
Chris@87:     x = np.matlib.rand(3)
Chris@87:     # check matrix type, array would have shape (3,)
Chris@87:     assert_(x.ndim == 2)
Chris@87: 
Chris@87: def test_randn():
Chris@87:     x = np.matlib.randn(3)
Chris@87:     # check matrix type, array would have shape (3,)
Chris@87:     assert_(x.ndim == 2)
Chris@87: 
Chris@87: def test_repmat():
Chris@87:     a1 = np.arange(4)
Chris@87:     x = np.matlib.repmat(a1, 2, 2)
Chris@87:     y = np.array([[0, 1, 2, 3, 0, 1, 2, 3],
Chris@87:                   [0, 1, 2, 3, 0, 1, 2, 3]])
Chris@87:     assert_array_equal(x, y)
Chris@87: 
Chris@87: 
Chris@87: if __name__ == "__main__":
Chris@87:     run_module_suite()