Examples/test-suite/python/python_pybuf_runme.py

# run:
#  python python_pybuf_runme.py benchmark
# for the benchmark, other wise the test case will be run
import python_pybuf
import sys


if len(sys.argv) >= 2 and sys.argv[1] == "benchmark":
    # run the benchmark
    import time
    k = 1000000  # number of times to execute the functions

    t = time.time()
    a = bytearray(b'hello world')
    for i in range(k):
        python_pybuf.title1(a)
    print "Time used by bytearray:", time.time() - t

    t = time.time()
    b = 'hello world'
    for i in range(k):
        python_pybuf.title2(b)
    print "Time used by string:", time.time() - t
else:
    # run the test case
    buf1 = bytearray(10)
    buf2 = bytearray(50)

    python_pybuf.func1(buf1)
    assert buf1 == b'a' * 10

    python_pybuf.func2(buf2)
    assert buf2.startswith(b"Hello world!\x00")

    count = python_pybuf.func3(buf2)
    assert count == 10  # number of alpha and number in 'Hello world!'

    length = python_pybuf.func4(buf2)
    assert length == 12

    buf3 = bytearray(b"hello")
    python_pybuf.title1(buf3)
    assert buf3 == b'Hello'