add testing for rmm tree