photoprism/pkg/colors/srgb_test.go

62 lines
926 B
Go
Raw Normal View History

package colors
import (
"image"
"image/jpeg"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/assert"
)
func writeImage(path string, img image.Image) error {
imgFile, err := os.Create(path)
if err != nil {
return err
}
defer imgFile.Close()
opt := jpeg.Options{
Quality: 95,
}
return jpeg.Encode(imgFile, img, &opt)
}
func TestToSRGB(t *testing.T) {
t.Run("DisplayP3", func(t *testing.T) {
testFile, _ := filepath.Abs("./testdata/DisplayP3.jpg")
t.Logf("testfile: %s", testFile)
imgFile, err := os.Open(testFile)
if err != nil {
t.Fatal(err)
}
defer imgFile.Close()
img, _, err := image.Decode(imgFile)
if err != nil {
t.Fatal(err)
}
imgSRGB := ToSRGB(img, ProfileDisplayP3)
srgbFile := "./testdata/SRGB.jpg"
if err := writeImage(srgbFile, imgSRGB); err != nil {
t.Error(err)
}
assert.FileExists(t, srgbFile)
_ = os.Remove(srgbFile)
})
}