ENH: adden new tests