ENH improve input checking
FIX api issues during tree fitting
ENH improve testing (assertions)