Add appropriate #[feature] directives to tests