MAINT: Added unittest for TDD