mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-27 19:30:20 +08:00
39 lines
821 B
Go
39 lines
821 B
Go
//go:build localtest
|
|
|
|
package ai
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetImageFromBuffer(t *testing.T) {
|
|
imagePath := "/Users/debugtalk/Downloads/s1.png"
|
|
file, err := os.ReadFile(imagePath)
|
|
require.Nil(t, err)
|
|
buf := new(bytes.Buffer)
|
|
buf.Read(file)
|
|
|
|
service := NewAIService(
|
|
WithCVService(CVServiceTypeVEDEM),
|
|
)
|
|
cvResult, err := service.ReadFromBuffer(buf)
|
|
assert.Nil(t, err)
|
|
fmt.Println(fmt.Sprintf("cvResult: %v", cvResult))
|
|
}
|
|
|
|
func TestGetImageFromPath(t *testing.T) {
|
|
imagePath := "/Users/debugtalk/Downloads/s1.png"
|
|
service := NewAIService(
|
|
WithCVService(CVServiceTypeVEDEM),
|
|
)
|
|
cvResult, err := service.ReadFromPath(imagePath)
|
|
assert.Nil(t, err)
|
|
fmt.Println(fmt.Sprintf("cvResult: %v", cvResult))
|
|
}
|