test: Add test for column_remove with index