Added unit test for forget_only_keys option