write_test function

Write test for an R function