Add more tests for io.go