Gecko [ goons4good.org ]

Name	Size	Permission	Date
test__datasource.py	9.91 KB	-rw-r--r--	2013-04-07 01:04
test__datasource.pyc	15.95 KB	-rw-r--r--	2018-04-10 19:40
test__datasource.pyo	15.95 KB	-rw-r--r--	2018-04-10 19:40
test__iotools.py	13.12 KB	-rw-r--r--	2013-04-07 01:04
test__iotools.pyc	14.4 KB	-rw-r--r--	2018-04-10 19:40
test__iotools.pyo	14.4 KB	-rw-r--r--	2018-04-10 19:40
test_arraypad.py	23.84 KB	-rw-r--r--	2013-04-07 01:04
test_arraypad.pyc	29.81 KB	-rw-r--r--	2018-04-10 19:40
test_arraypad.pyo	29.81 KB	-rw-r--r--	2018-04-10 19:40
test_arraysetops.py	7.52 KB	-rw-r--r--	2013-04-07 01:04
test_arraysetops.pyc	7.88 KB	-rw-r--r--	2018-04-10 19:40
test_arraysetops.pyo	7.88 KB	-rw-r--r--	2018-04-10 19:40
test_arrayterator.py	1.4 KB	-rw-r--r--	2013-04-07 01:04
test_arrayterator.pyc	1.98 KB	-rw-r--r--	2018-04-10 19:40
test_arrayterator.pyo	1.98 KB	-rw-r--r--	2018-04-10 19:40
test_financial.py	5.32 KB	-rw-r--r--	2013-04-07 01:04
test_financial.pyc	6.21 KB	-rw-r--r--	2018-04-10 19:40
test_financial.pyo	6.21 KB	-rw-r--r--	2018-04-10 19:40
test_format.py	25.04 KB	-rw-r--r--	2013-04-07 01:04
test_format.pyc	25 KB	-rw-r--r--	2018-04-10 19:40
test_format.pyo	25 KB	-rw-r--r--	2018-04-10 19:40
test_function_base.py	50.27 KB	-rw-r--r--	2013-04-07 01:04
test_function_base.pyc	72.44 KB	-rw-r--r--	2018-04-10 19:40
test_function_base.pyo	72.44 KB	-rw-r--r--	2018-04-10 19:40
test_index_tricks.py	8.98 KB	-rw-r--r--	2013-04-07 01:04
test_index_tricks.pyc	12.25 KB	-rw-r--r--	2018-04-10 19:40
test_index_tricks.pyo	12.25 KB	-rw-r--r--	2018-04-10 19:40
test_io.py	59.81 KB	-rw-r--r--	2013-04-07 01:04
test_io.pyc	73.17 KB	-rw-r--r--	2018-04-10 19:40
test_io.pyo	73.17 KB	-rw-r--r--	2018-04-10 19:40
test_polynomial.py	4.41 KB	-rw-r--r--	2013-04-07 01:04
test_polynomial.pyc	6.1 KB	-rw-r--r--	2018-04-10 19:40
test_polynomial.pyo	6.1 KB	-rw-r--r--	2018-04-10 19:40
test_recfunctions.py	27.93 KB	-rw-r--r--	2013-04-07 01:04
test_recfunctions.pyc	35.45 KB	-rw-r--r--	2018-04-10 19:40
test_recfunctions.pyo	35.45 KB	-rw-r--r--	2018-04-10 19:40
test_regression.py	8.02 KB	-rw-r--r--	2013-04-07 01:04
test_regression.pyc	13.26 KB	-rw-r--r--	2018-04-10 19:40
test_regression.pyo	13.26 KB	-rw-r--r--	2018-04-10 19:40
test_shape_base.py	9.92 KB	-rw-r--r--	2013-04-07 01:04
test_shape_base.pyc	15.89 KB	-rw-r--r--	2018-04-10 19:40
test_shape_base.pyo	15.89 KB	-rw-r--r--	2018-04-10 19:40
test_stride_tricks.py	6.79 KB	-rw-r--r--	2013-04-07 01:04
test_stride_tricks.pyc	8.92 KB	-rw-r--r--	2018-04-10 19:40
test_stride_tricks.pyo	8.92 KB	-rw-r--r--	2018-04-10 19:40
test_twodim_base.py	11.38 KB	-rw-r--r--	2013-04-07 01:04
test_twodim_base.pyc	15.93 KB	-rw-r--r--	2018-04-10 19:40
test_twodim_base.pyo	15.93 KB	-rw-r--r--	2018-04-10 19:40
test_type_check.py	10.33 KB	-rw-r--r--	2013-04-07 01:04
test_type_check.pyc	18.71 KB	-rw-r--r--	2018-04-10 19:40
test_type_check.pyo	18.71 KB	-rw-r--r--	2018-04-10 19:40
test_ufunclike.py	1.88 KB	-rw-r--r--	2013-04-07 01:04
test_ufunclike.pyc	3.33 KB	-rw-r--r--	2018-04-10 19:40
test_ufunclike.pyo	3.33 KB	-rw-r--r--	2018-04-10 19:40
test_utils.py	920 B	-rw-r--r--	2013-04-07 01:04
test_utils.pyc	2.13 KB	-rw-r--r--	2018-04-10 19:40
test_utils.pyo	2.13 KB	-rw-r--r--	2018-04-10 19:40

Rename

import numpy as np
from numpy.testing import *
from numpy.lib.stride_tricks import broadcast_arrays

def assert_shapes_correct(input_shapes, expected_shape):
    """ Broadcast a list of arrays with the given input shapes and check the
    common output shape.
    """
    inarrays = [np.zeros(s) for s in input_shapes]
    outarrays = broadcast_arrays(*inarrays)
    outshapes = [a.shape for a in outarrays]
    expected = [expected_shape] * len(inarrays)
    assert_equal(outshapes, expected)

def assert_incompatible_shapes_raise(input_shapes):
    """ Broadcast a list of arrays with the given (incompatible) input shapes
    and check that they raise a ValueError.
    """
    inarrays = [np.zeros(s) for s in input_shapes]
    assert_raises(ValueError, broadcast_arrays, *inarrays)

def assert_same_as_ufunc(shape0, shape1, transposed=False, flipped=False):
    """ Broadcast two shapes against each other and check that the data layout
    is the same as if a ufunc did the broadcasting.
    """
    x0 = np.zeros(shape0, dtype=int)
    # Note that multiply.reduce's identity element is 1.0, so when shape1==(),
    # this gives the desired n==1.
    n = int(np.multiply.reduce(shape1))
    x1 = np.arange(n).reshape(shape1)
    if transposed:
        x0 = x0.T
        x1 = x1.T
    if flipped:
        x0 = x0[::-1]
        x1 = x1[::-1]
    # Use the add ufunc to do the broadcasting. Since we're adding 0s to x1, the
    # result should be exactly the same as the broadcasted view of x1.
    y = x0 + x1
    b0, b1 = broadcast_arrays(x0, x1)
    assert_array_equal(y, b1)

def test_same():
    x = np.arange(10)
    y = np.arange(10)
    bx, by = broadcast_arrays(x, y)
    assert_array_equal(x, bx)
    assert_array_equal(y, by)

def test_one_off():
    x = np.array([[1,2,3]])
    y = np.array([[1],[2],[3]])
    bx, by = broadcast_arrays(x, y)
    bx0 = np.array([[1,2,3],[1,2,3],[1,2,3]])
    by0 = bx0.T
    assert_array_equal(bx0, bx)
    assert_array_equal(by0, by)

def test_same_input_shapes():
    """ Check that the final shape is just the input shape.
    """
    data = [
        (),
        (1,),
        (3,),
        (0,1),
        (0,3),
        (1,0),
        (3,0),
        (1,3),
        (3,1),
        (3,3),
    ]
    for shape in data:
        input_shapes = [shape]
        # Single input.
        assert_shapes_correct(input_shapes, shape)
        # Double input.
        input_shapes2 = [shape, shape]
        assert_shapes_correct(input_shapes2, shape)
        # Triple input.
        input_shapes3 = [shape, shape, shape]
        assert_shapes_correct(input_shapes3, shape)

def test_two_compatible_by_ones_input_shapes():
    """ Check that two different input shapes (of the same length but some have
    1s) broadcast to the correct shape.
    """
    data = [
        [[(1,), (3,)], (3,)],
        [[(1,3), (3,3)], (3,3)],
        [[(3,1), (3,3)], (3,3)],
        [[(1,3), (3,1)], (3,3)],
        [[(1,1), (3,3)], (3,3)],
        [[(1,1), (1,3)], (1,3)],
        [[(1,1), (3,1)], (3,1)],
        [[(1,0), (0,0)], (0,0)],
        [[(0,1), (0,0)], (0,0)],
        [[(1,0), (0,1)], (0,0)],
        [[(1,1), (0,0)], (0,0)],
        [[(1,1), (1,0)], (1,0)],
        [[(1,1), (0,1)], (0,1)],
    ]
    for input_shapes, expected_shape in data:
        assert_shapes_correct(input_shapes, expected_shape)
        # Reverse the input shapes since broadcasting should be symmetric.
        assert_shapes_correct(input_shapes[::-1], expected_shape)

def test_two_compatible_by_prepending_ones_input_shapes():
    """ Check that two different input shapes (of different lengths) broadcast
    to the correct shape.
    """
    data = [
        [[(), (3,)], (3,)],
        [[(3,), (3,3)], (3,3)],
        [[(3,), (3,1)], (3,3)],
        [[(1,), (3,3)], (3,3)],
        [[(), (3,3)], (3,3)],
        [[(1,1), (3,)], (1,3)],
        [[(1,), (3,1)], (3,1)],
        [[(1,), (1,3)], (1,3)],
        [[(), (1,3)], (1,3)],
        [[(), (3,1)], (3,1)],
        [[(), (0,)], (0,)],
        [[(0,), (0,0)], (0,0)],
        [[(0,), (0,1)], (0,0)],
        [[(1,), (0,0)], (0,0)],
        [[(), (0,0)], (0,0)],
        [[(1,1), (0,)], (1,0)],
        [[(1,), (0,1)], (0,1)],
        [[(1,), (1,0)], (1,0)],
        [[(), (1,0)], (1,0)],
        [[(), (0,1)], (0,1)],
    ]
    for input_shapes, expected_shape in data:
        assert_shapes_correct(input_shapes, expected_shape)
        # Reverse the input shapes since broadcasting should be symmetric.
        assert_shapes_correct(input_shapes[::-1], expected_shape)

def test_incompatible_shapes_raise_valueerror():
    """ Check that a ValueError is raised for incompatible shapes.
    """
    data = [
        [(3,), (4,)],
        [(2,3), (2,)],
        [(3,), (3,), (4,)],
        [(1,3,4), (2,3,3)],
    ]
    for input_shapes in data:
        assert_incompatible_shapes_raise(input_shapes)
        # Reverse the input shapes since broadcasting should be symmetric.
        assert_incompatible_shapes_raise(input_shapes[::-1])

def test_same_as_ufunc():
    """ Check that the data layout is the same as if a ufunc did the operation.
    """
    data = [
        [[(1,), (3,)], (3,)],
        [[(1,3), (3,3)], (3,3)],
        [[(3,1), (3,3)], (3,3)],
        [[(1,3), (3,1)], (3,3)],
        [[(1,1), (3,3)], (3,3)],
        [[(1,1), (1,3)], (1,3)],
        [[(1,1), (3,1)], (3,1)],
        [[(1,0), (0,0)], (0,0)],
        [[(0,1), (0,0)], (0,0)],
        [[(1,0), (0,1)], (0,0)],
        [[(1,1), (0,0)], (0,0)],
        [[(1,1), (1,0)], (1,0)],
        [[(1,1), (0,1)], (0,1)],
        [[(), (3,)], (3,)],
        [[(3,), (3,3)], (3,3)],
        [[(3,), (3,1)], (3,3)],
        [[(1,), (3,3)], (3,3)],
        [[(), (3,3)], (3,3)],
        [[(1,1), (3,)], (1,3)],
        [[(1,), (3,1)], (3,1)],
        [[(1,), (1,3)], (1,3)],
        [[(), (1,3)], (1,3)],
        [[(), (3,1)], (3,1)],
        [[(), (0,)], (0,)],
        [[(0,), (0,0)], (0,0)],
        [[(0,), (0,1)], (0,0)],
        [[(1,), (0,0)], (0,0)],
        [[(), (0,0)], (0,0)],
        [[(1,1), (0,)], (1,0)],
        [[(1,), (0,1)], (0,1)],
        [[(1,), (1,0)], (1,0)],
        [[(), (1,0)], (1,0)],
        [[(), (0,1)], (0,1)],
    ]
    for input_shapes, expected_shape in data:
        assert_same_as_ufunc(input_shapes[0], input_shapes[1],
                "Shapes: %s %s" % (input_shapes[0], input_shapes[1]))
        # Reverse the input shapes since broadcasting should be symmetric.
        assert_same_as_ufunc(input_shapes[1], input_shapes[0])
        # Try them transposed, too.
        assert_same_as_ufunc(input_shapes[0], input_shapes[1], True)
        # ... and flipped for non-rank-0 inputs in order to test negative
        # strides.
        if () not in input_shapes:
            assert_same_as_ufunc(input_shapes[0], input_shapes[1], False, True)
            assert_same_as_ufunc(input_shapes[0], input_shapes[1], True, True)

if __name__ == "__main__":
    run_module_suite()