Fix broken test case for `get_extra`

Use TOTAL_FORMS count. It'll remain 2 for a binary tree

Signed-off-by: Rohan Jain <crodjer@gmail.com>