Python print_string_array示例

示例#1

文件： test_numpy_dtypes.py 项目： rhl-/random_forest

def test_string_array():
    from pybind11_tests import create_string_array, print_string_array

    arr = create_string_array(True)
    assert str(arr.dtype) == "[('a', 'S3'), ('b', 'S3')]"
    assert print_string_array(arr) == [
        "a='',b=''", "a='a',b='a'", "a='ab',b='ab'", "a='abc',b='abc'"
    ]
    dtype = arr.dtype
    assert arr['a'].tolist() == [b'', b'a', b'ab', b'abc']
    assert arr['b'].tolist() == [b'', b'a', b'ab', b'abc']
    arr = create_string_array(False)
    assert dtype == arr.dtype

示例#2

文件： test_numpy_dtypes.py 项目： aldanor/pybind11

def test_string_array():
    from pybind11_tests import create_string_array, print_string_array

    arr = create_string_array(True)
    assert str(arr.dtype) == "[('a', 'S3'), ('b', 'S3')]"
    assert print_string_array(arr) == [
        "a='',b=''",
        "a='a',b='a'",
        "a='ab',b='ab'",
        "a='abc',b='abc'"
    ]
    dtype = arr.dtype
    assert arr['a'].tolist() == [b'', b'a', b'ab', b'abc']
    assert arr['b'].tolist() == [b'', b'a', b'ab', b'abc']
    arr = create_string_array(False)
    assert dtype == arr.dtype

示例#3

def test_string_array():
    from pybind11_tests import create_string_array, print_string_array

    arr = create_string_array(True)
    if str(arr.dtype) != "[('a', 'S3'), ('b', 'S3')]":
        raise AssertionError
    if print_string_array(arr) != [
            "a='',b=''", "a='a',b='a'", "a='ab',b='ab'", "a='abc',b='abc'"
    ]:
        raise AssertionError
    dtype = arr.dtype
    if arr['a'].tolist() != [b'', b'a', b'ab', b'abc']:
        raise AssertionError
    if arr['b'].tolist() != [b'', b'a', b'ab', b'abc']:
        raise AssertionError
    arr = create_string_array(False)
    if dtype != arr.dtype:
        raise AssertionError