added unit tests for tree_height()

This closes #54.