Move tests out of src and into test