Write some more test cases for collect