ENH: Use decorator to speed up tests