Move tests to src