Format test code with gofmt