ENH: Add test case