Move test code to test/