示例#1
0
def test_func2():
    assert_equal(
        X('[1, ...]', '[fun(1), ...]')(np.r_[1, 2, 3],
                                       f={
                                           'fun': lambda x: x * 10
                                       }), np.r_[10, 20, 30])
    assert_equal(
        X('[1, ...]', '[fun(1), ...]')(np.r_[1, 2, 3],
                                       f={
                                           'fun': lambda x: x * 5
                                       }), np.r_[5, 10, 15])
示例#2
0
def test_2out():
    a = X('[1, 2, ...]; [a, b, ...]',
          '[1, a, 2, b, ...]; [[a, 1], [b, 2], ...]')
    assert_equal(a[0]([np.r_[1, 2, 3, 4], np.r_[-1, -2]]), np.r_[1, -1, 2, -2,
                                                                 3])
    assert_equal(a[1]([np.r_[1, 2, 3, 4], np.r_[-1, -2]]),
                 np.array([[-1, 1], [-2, 2]]))
示例#3
0
def test_extraShape2():
    assert_equal(
        X('[1, 2, 3, ...]',
          '[[1, 2], [2, 3], ...]; [[1, 2, 3], ...]')(np.r_[0, 1, 2, 3],
                                                     extraShapes=[(1, 0),
                                                                  (1, 0)]),
        (np.array([[0, 1], [1, 2], [2, 3], [3, 0]
                   ]), np.array([[0, 1, 2], [1, 2, 3], [2, 3, 0]])))
示例#4
0
def test_2in_list():
    assert_equal(
        X(['[1, 2, ...]', '[a, b, ...]'],
          '[1, a, 2, b, ...]')([np.r_[1, 2, 3, 4], np.r_[-1, -2, -3]]),
        np.r_[1, -1, 2, -2, 3, -3, 4])
示例#5
0
def test_2in_outShape_label():
    assert_equal(
        X('[a0, b0, a1, b1, ...]')([np.r_[1, 2, 3, 4], np.r_[-1, -2, -3]]),
        np.r_[1, -1, 2, -2, 3, -3, 4])
示例#6
0
def test_ZeroIndexConverter_1in():
    assert_equal(X('1', '[1, 1]')(np.r_[1, 2]), np.array([[1, 2], [1, 2]]))
示例#7
0
def test_starred():
    assert_equal(
        X('[1, ...]', '[*1, ...]')(np.array([[1, 2], [2, 3], [3, 4]])),
        np.array([1, 2, 2, 3, 3, 4]))
示例#8
0
def test_starred_label():
    assert_equal(
        X('[*a0, ...]')(np.array([[1, 2], [2, 3], [3, 4]])),
        np.array([1, 2, 2, 3, 3, 4]))
示例#9
0
def test_starred_2in():
    assert_equal(
        X('1;2', '[*1, *2]')([np.r_[1, 2], np.r_[3, 4]]),
        np.array([1, 2, 3, 4]))
示例#10
0
def test_starred_2in_label():
    assert_equal(
        X('[*a, *b]')([np.r_[1, 2], np.r_[3, 4]]), np.array([1, 2, 3, 4]))
示例#11
0
def test_2in_1use():
    assert_equal(
        X('[1, 2, ...]; [a, b, ...]', '[[1, 1], [2, 2], ...]')(
            [np.array([1, 2, 3, 4]),
             np.array([[-1, -2], [-3, -4], [-5, -6]])]),
        np.array([[1, 1], [2, 2], [3, 3], [4, 4]]))
示例#12
0
def test_2out_label():
    a = X('[a0, b0, a1, b1, ...]; [[b0, a0], [b1, a1], ...]')
    assert_equal(a[0]([np.r_[1, 2, 3, 4], np.r_[-1, -2]]), np.r_[1, -1, 2, -2,
                                                                 3])
    assert_equal(a[1]([np.r_[1, 2, 3, 4], np.r_[-1, -2]]),
                 np.array([[-1, 1], [-2, 2]]))
示例#13
0
def test_outShape3():
    assert_equal(
        X('[[a00, a01], ...]')(np.array([[1, 2], [3, 4]])),
        np.array([[1, 2], [3, 4]]))
示例#14
0
def test_2in_outShape2():
    assert_equal(
        X('[1, 2, ...]; [a, b, ...]',
          '[1, a, 2, b, ...]')([np.r_[1, 2, 3, 4], np.r_[-1, -2]]),
        np.r_[1, -1, 2, -2, 3])
示例#15
0
def test_outShape2():
    assert_equal(
        X('[1, ...]', '[[1, 1, ...], ...]')(np.r_[1, 2, 3],
                                            outShapes=[(-1, 2)]),
        np.array([[1, 1], [2, 2], [3, 3]]))
示例#16
0
def test_func3_label():
    assert_equal(
        X('[[fun(a00), a01], ...]', f={'fun':
                                       lambda x: x * 10})(np.array([[1, 2],
                                                                    [3, 4]])),
        np.array([[10, 2], [30, 4]]))
示例#17
0
def test_ZeroIndexConverter_2in():
    assert_equal(
        X('1;2', '[1, 2]')([np.r_[1, 2], np.r_[3, 4]]),
        np.array([[1, 2], [3, 4]]))
示例#18
0
def test_func3():
    assert_equal(
        X('[[1, 2], ...]', '[[fun(1), 2], ...]',
          f={'fun': lambda x: x * 10})(np.array([[1, 2], [3, 4]])),
        np.array([[10, 2], [30, 4]]))
示例#19
0
def test_func():
    assert_equal(
        X('[1, ...]', '[fun(1), ...]',
          f={'fun': lambda x: x * 10})(np.r_[1, 2, 3]), np.r_[10, 20, 30])
示例#20
0
def test_2in_subarray():
    assert_equal(
        X('[1, 2, ...]; [a, b, ...]', '[1, a, 2, b, ...]')([
            np.array([[1, 2], [3, 4]]),
            np.array([[-1, -2], [-3, -4], [-5, -6]])
        ]), np.array([[1, 2], [-1, -2], [3, 4], [-3, -4]]))
示例#21
0
def test_ZeroIndexConverter_2in_label():
    assert_equal(
        X('[a, b]')([np.r_[1, 2], np.r_[3, 4]]), np.array([[1, 2], [3, 4]]))
示例#22
0
def test_outShape5():
    assert_equal(
        X('[[*a, b0], ...]')([np.r_[1, 2, 3], np.r_[10, 20, 30, 40]]),
        np.array([[1, 2, 3, 10], [1, 2, 3, 20], [1, 2, 3, 30], [1, 2, 3, 40]]))
示例#23
0
def test_ZeroIndexConverter_1in_label():
    assert_equal(X('[a, a]')(np.r_[1, 2]), np.array([[1, 2], [1, 2]]))