Add tests for head/sortWithTake