ENH: adding tests