added unit tests for tree_height()

This closes #54.
Merge pull request #56 from drincruz/feature/unit_tests/tree_height

added unit tests for tree_height()