Python mutate_at_tの例

コード例 #1

ファイル: test_numpy_array.py プロジェクト: HDembinski/pybind11

def test_at(arr):
    from pybind11_tests.array import at_t, mutate_at_t

    assert at_t(arr, 0, 2) == 3
    assert at_t(arr, 1, 0) == 4

    assert all(mutate_at_t(arr, 0, 2).ravel() == [1, 2, 4, 4, 5, 6])
    assert all(mutate_at_t(arr, 1, 0).ravel() == [1, 2, 4, 5, 5, 6])

コード例 #2

def test_at(arr):
    from pybind11_tests.array import at_t, mutate_at_t

    assert at_t(arr, 0, 2) == 3
    assert at_t(arr, 1, 0) == 4

    assert all(mutate_at_t(arr, 0, 2).ravel() == [1, 2, 4, 4, 5, 6])
    assert all(mutate_at_t(arr, 1, 0).ravel() == [1, 2, 4, 5, 5, 6])

コード例 #3

def test_at(arr):
    from pybind11_tests.array import at_t, mutate_at_t

    if at_t(arr, 0, 2) != 3:
        raise AssertionError
    if at_t(arr, 1, 0) != 4:
        raise AssertionError

    if not all(mutate_at_t(arr, 0, 2).ravel() == [1, 2, 4, 4, 5, 6]):
        raise AssertionError
    if not all(mutate_at_t(arr, 1, 0).ravel() == [1, 2, 4, 5, 5, 6]):
        raise AssertionError