ENH: Updated tests.