/trunk/Examples/test-suite/python/python_pybuf_runme3.py

 1#run: 
 2#  python python_pybuf_runme3.py benchmark
 3#for the benchmark, other wise the test case will be run
 4import python_pybuf
 5import sys
 6if len(sys.argv)>=2 and sys.argv[1]=="benchmark":
 7    #run the benchmark
 8    import time
 9    k=1000000 #number of times to excute the functions
10
11    t=time.time()
12    a = bytearray(b'hello world')
13    for i in range(k):
14      pybuf.title1(a)
15    print("Time used by bytearray:",time.time()-t)
16
17    t=time.time()
18    b = 'hello world'
19    for i in range(k):
20      pybuf.title2(b)
21    print("Time used by string:",time.time()-t)
22else:
23    #run the test case
24    buf1 = bytearray(10)
25    buf2 = bytearray(50)
26
27    pybuf.func1(buf1)
28    assert buf1 == b'a'*10
29
30    pybuf.func2(buf2)
31    assert buf2.startswith(b"Hello world!\x00")
32
33    count = pybuf.func3(buf2)
34    assert count==10 #number of alpha and number in 'Hello world!'
35
36    length = pybuf.func4(buf2)
37    assert length==12
38
39    buf3 = bytearray(b"hello")
40    pybuf.title1(buf3)
41    assert buf3==b'Hello'
42