116 lines
3.2 KiB
Go
116 lines
3.2 KiB
Go
package images4
|
|
|
|
import (
|
|
"path"
|
|
"testing"
|
|
)
|
|
|
|
func testPropSimilar(fA, fB string, isSimilar bool,
|
|
t *testing.T) {
|
|
p := path.Join("testdata", "proportions")
|
|
imgA, err := Open(path.Join(p, fA))
|
|
if err != nil {
|
|
t.Error("Error opening image:", err)
|
|
}
|
|
imgB, err := Open(path.Join(p, fB))
|
|
if err != nil {
|
|
t.Error("Error opening image:", err)
|
|
}
|
|
iconA := Icon(imgA)
|
|
iconB := Icon(imgB)
|
|
|
|
if isSimilar == true {
|
|
if !propSimilar(iconA, iconB) {
|
|
t.Errorf("Expecting similarity of %v to %v.", fA, fB)
|
|
}
|
|
}
|
|
if isSimilar == false {
|
|
if propSimilar(iconA, iconB) {
|
|
t.Errorf("Expecting non-similarity of %v to %v.", fA, fB)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPropSimilar(t *testing.T) {
|
|
testPropSimilar("100x130.png", "100x124.png", true, t)
|
|
testPropSimilar("100x130.png", "100x122.png", false, t)
|
|
testPropSimilar("130x100.png", "260x200.png", true, t)
|
|
testPropSimilar("200x200.png", "260x200.png", false, t)
|
|
testPropSimilar("130x100.png", "124x100.png", true, t)
|
|
testPropSimilar("130x100.png", "122x100.png", false, t)
|
|
testPropSimilar("130x100.png", "130x100.png", true, t)
|
|
testPropSimilar("100x130.png", "130x100.png", false, t)
|
|
testPropSimilar("124x100.png", "260x200.png", true, t)
|
|
testPropSimilar("122x100.png", "260x200.png", false, t)
|
|
testPropSimilar("100x124.png", "100x130.png", true, t)
|
|
}
|
|
|
|
func testEucSimilar(fA, fB string, isSimilar bool,
|
|
t *testing.T) {
|
|
p := path.Join("testdata", "euclidean")
|
|
imgA, err := Open(path.Join(p, fA))
|
|
if err != nil {
|
|
t.Error("Error opening image:", err)
|
|
}
|
|
iconA := Icon(imgA)
|
|
imgB, err := Open(path.Join(p, fB))
|
|
if err != nil {
|
|
t.Error("Error opening image:", err)
|
|
}
|
|
iconB := Icon(imgB)
|
|
if isSimilar == true {
|
|
if !eucSimilar(iconA, iconB) {
|
|
t.Errorf("Expecting similarity of %v to %v.", fA, fB)
|
|
}
|
|
}
|
|
if isSimilar == false {
|
|
if eucSimilar(iconA, iconB) {
|
|
t.Errorf("Expecting non-similarity of %v to %v.", fA, fB)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEucSimilar(t *testing.T) {
|
|
testEucSimilar("large.jpg", "distorted.jpg", true, t)
|
|
testEucSimilar("large.jpg", "flipped.jpg", false, t)
|
|
testEucSimilar("large.jpg", "small.jpg", true, t)
|
|
testEucSimilar("small.gif", "small.jpg", true, t) // GIF test.
|
|
testEucSimilar("uniform-black.png", "uniform-black.png", true, t)
|
|
testEucSimilar("uniform-black.png", "uniform-white.png", false, t)
|
|
testEucSimilar("uniform-green.png", "uniform-green.png", true, t)
|
|
testEucSimilar("uniform-green.png", "uniform-white.png", false, t)
|
|
testEucSimilar("uniform-white.png", "uniform-white.png", true, t)
|
|
}
|
|
|
|
func TestSimilar90270(t *testing.T) {
|
|
|
|
img0, _ := Open(path.Join("testdata", "rotate", "0.jpg"))
|
|
img90, _ := Open(path.Join("testdata", "rotate", "90.jpg"))
|
|
img180, _ := Open(path.Join("testdata", "rotate", "180.jpg"))
|
|
img270, _ := Open(path.Join("testdata", "rotate", "270.jpg"))
|
|
|
|
icon0 := Icon(img0)
|
|
icon90 := Icon(img90)
|
|
icon180 := Icon(img180)
|
|
icon270 := Icon(img270)
|
|
|
|
if !Similar90270(icon0, icon90) {
|
|
t.Errorf("0.jpg must be similar to 90.jpg")
|
|
}
|
|
if Similar90270(icon0, icon180) {
|
|
t.Errorf("0.jpg must be NOT similar to 180.jpg")
|
|
}
|
|
|
|
if !Similar90270(icon0, icon270) {
|
|
t.Errorf("0.jpg must be similar to 270.jpg")
|
|
}
|
|
|
|
if !Similar90270(icon90, icon180) {
|
|
t.Errorf("90.jpg must be similar to 180.jpg")
|
|
}
|
|
|
|
if Similar90270(icon90, icon270) {
|
|
t.Errorf("90.jpg must be NOT similar to 270.jpg")
|
|
}
|
|
}
|