ENH added tests