add unit tests for `nextSubtreeSize`

[rach-id]

          Is this func `nextSubtreeSize` unit tested? if not, I'd suggest adding unit tests for it in order to ensure its behaviour is aligned with what it is intended for.

Originally posted by @staheri14 in #260 (comment)