38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from numpy.testing import assert_raises
|
|
import numpy as np
|
|
|
|
from .. import all
|
|
from .._creation_functions import asarray
|
|
from .._dtypes import float64, int8
|
|
from .._manipulation_functions import (
|
|
concat,
|
|
reshape,
|
|
stack
|
|
)
|
|
|
|
|
|
def test_concat_errors():
|
|
assert_raises(TypeError, lambda: concat((1, 1), axis=None))
|
|
assert_raises(TypeError, lambda: concat([asarray([1], dtype=int8),
|
|
asarray([1], dtype=float64)]))
|
|
|
|
|
|
def test_stack_errors():
|
|
assert_raises(TypeError, lambda: stack([asarray([1, 1], dtype=int8),
|
|
asarray([2, 2], dtype=float64)]))
|
|
|
|
|
|
def test_reshape_copy():
|
|
a = asarray(np.ones((2, 3)))
|
|
b = reshape(a, (3, 2), copy=True)
|
|
assert not np.shares_memory(a._array, b._array)
|
|
|
|
a = asarray(np.ones((2, 3)))
|
|
b = reshape(a, (3, 2), copy=False)
|
|
assert np.shares_memory(a._array, b._array)
|
|
|
|
a = asarray(np.ones((2, 3)).T)
|
|
b = reshape(a, (3, 2), copy=True)
|
|
assert_raises(AttributeError, lambda: reshape(a, (2, 3), copy=False))
|
|
|