add test function

add test function