Moved tests from src to module