Improve IO class unit testing