Writing your own Test unit