Add tests for animal.py