The tests in this file have a lot of duplication (e.g. around resetting the stream at the end). It would be helpful to go through and extract this into helper functions to improve readability