Organize import statements in test classes.