Add to ablation tests to check ablated_features columns