ENH: Add test helper