TST proper testing import
ENH simplify input testing