Move unit tests to its package