Add tests for the util.go file.