[test] Added tests for .remove