ENH: add test