REDROOM

Edit File Size: 1.58 KB

/opt/alt/python37/lib64/python3.7/site-packages/numpy/tests/test_matlib.py

Text Base64

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 = numpy.matlib.empty((2,))
    assert_(isinstance(x, np.matrix))
    assert_(x.shape, (1, 2))

def test_ones():
    assert_array_equal(numpy.matlib.ones((2, 3)),
                       np.matrix([[ 1.,  1.,  1.],
                                 [ 1.,  1.,  1.]]))

assert_array_equal(numpy.matlib.ones(2), np.matrix([[ 1.,  1.]]))

def test_zeros():
    assert_array_equal(numpy.matlib.zeros((2, 3)),
                       np.matrix([[ 0.,  0.,  0.],
                                 [ 0.,  0.,  0.]]))

assert_array_equal(numpy.matlib.zeros(2), np.matrix([[ 0.,  0.]]))

def test_identity():
    x = numpy.matlib.identity(2, dtype=np.int)
    assert_array_equal(x, np.matrix([[1, 0], [0, 1]]))

def test_eye():
    x = numpy.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 = numpy.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 = numpy.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()

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2023-05-13 02:18:14	Edit Download recursive recursive
test_ctypeslib.py	4.35 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_matlib.py	1.58 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_numpy_version.py	717 B	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_reloading.py	1.01 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_scripts.py	3.01 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_warnings.py	2.66 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive

Directory Contents