Add unit tests for Collection element extract