Add a test for vec::retain