98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
package collage
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
"image"
|
|
"image/jpeg"
|
|
"io"
|
|
"io/fs"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
//go:embed test_data/*
|
|
var testData embed.FS
|
|
|
|
var testDataFS fs.FS
|
|
|
|
func init() {
|
|
var err error
|
|
testDataFS, err = fs.Sub(testData, "test_data")
|
|
if err != nil {
|
|
log.Panicf("getting test_data subdir failed %v", err)
|
|
}
|
|
}
|
|
|
|
func TestMake(t *testing.T) {
|
|
t.Log("Test was run")
|
|
img1f, err := testDataFS.Open("img1.jpg")
|
|
if err != nil {
|
|
t.Fatalf("opening image failed %v", err)
|
|
}
|
|
fmt.Printf("%#v\n", img1f)
|
|
}
|
|
|
|
func TestCrop(t *testing.T) {
|
|
img, err := GetImage(testDataFS, "img1.jpg")
|
|
if err != nil {
|
|
t.Fatalf("failed to get image %v", err)
|
|
}
|
|
r := Rectangle{
|
|
Start: Point{X: 431, Y: 697},
|
|
End: Point{X: 2514, Y: 2047},
|
|
}
|
|
cropped, err := Crop(img, r)
|
|
if err != nil {
|
|
t.Fatalf("failed to crop image %v", err)
|
|
}
|
|
|
|
croppedImgPrefix := fmt.Sprint(cropped)[:1000]
|
|
|
|
refPrefixf, err := testDataFS.Open("cropped_image_prefix.txt")
|
|
if err != nil {
|
|
t.Fatalf("failed to open ref prefix %v", err)
|
|
}
|
|
|
|
refImgPrefix, err := io.ReadAll(refPrefixf)
|
|
|
|
if err != nil {
|
|
t.Fatalf("failed to read ref prefix %v", err)
|
|
}
|
|
|
|
if string(refImgPrefix) != croppedImgPrefix {
|
|
t.Fatalf("Cropped image is not the same as reference image")
|
|
}
|
|
// SavePrefix(cropped)
|
|
// expectedImage, err := GetImage(testDataFS, "test_output.jpg")
|
|
// if err != nil {
|
|
// t.Fatalf("failed to get reference crop image %v", err)
|
|
// }
|
|
// fmt.Printf("%v\n", cropped)
|
|
// fmt.Printf("%v\n", expectedImage)
|
|
// if fmt.Sprintf("%#v", cropped) != fmt.Sprintf("%#v", expectedImage) {
|
|
// t.Fatalf("Cropped image is not the same as reference image")
|
|
// }
|
|
// SaveImage(cropped)
|
|
}
|
|
|
|
// Save first 1000 bytes of string representation to compare against reference
|
|
func SavePrefix(img image.Image) error {
|
|
imgStr := fmt.Sprint(img)
|
|
prefix := imgStr[:1000]
|
|
return os.WriteFile("./test_data_prefix.txt", []byte(prefix), 0644)
|
|
}
|
|
|
|
func SaveImage(img image.Image) {
|
|
out, err := os.Create("./test_output.jpg")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
var opt jpeg.Options
|
|
opt.Quality = 100
|
|
|
|
jpeg.Encode(out, img, &opt)
|
|
}
|