62 lines
926 B
Go
62 lines
926 B
Go
|
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)
|
||
|
|
})
|
||
|
|
}
|