Add unit tests for Heterogeneous tree