Moving test to inference folder