adding testthat tests