fix: unittest

This commit is contained in:
xucong.053
2023-04-19 19:29:36 +08:00
parent 59e82cef72
commit c4cf4c9d83
2 changed files with 5 additions and 2 deletions
+4 -1
View File
@@ -64,7 +64,10 @@ type ImageTemplate struct {
}
func NewImageTemplate(filepath string) ImageTemplate {
return ImageTemplate{FilePath: filepath}
return ImageTemplate{
FilePath: filepath,
rawBuffer: bytes.NewBuffer([]byte{}),
}
}
func (i *ImageTemplate) read() (imageBuffer *bytes.Buffer, err error) {
+1 -1
View File
@@ -58,7 +58,7 @@ func checkIMEnv() error {
}
func convertBase64(imgByte []byte) (baseImg string) {
return base64.StdEncoding.EncodeToString(imgByte)
return base64.RawStdEncoding.EncodeToString(imgByte)
}
func (s *veDEMIMService) getIMResult(searchImage []byte, sourceImage []byte) ([]IMResult, error) {