127 lines
3.1 KiB
Go
127 lines
3.1 KiB
Go
// Copyright 2017 The goimagehash Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package goimagehash
|
|
|
|
import (
|
|
"errors"
|
|
"image"
|
|
_ "image/jpeg"
|
|
"os"
|
|
"reflect"
|
|
"runtime"
|
|
"testing"
|
|
)
|
|
|
|
func TestNewImageHash(t *testing.T) {
|
|
for _, tt := range []struct {
|
|
datas [][]uint8
|
|
hash1 Kind
|
|
hash2 Kind
|
|
distance int
|
|
err error
|
|
}{
|
|
{[][]uint8{{1, 0, 1, 1}, {0, 0, 0, 0}}, Unknown, Unknown, 3, nil},
|
|
{[][]uint8{{0, 0, 0, 0}, {0, 0, 0, 0}}, Unknown, Unknown, 0, nil},
|
|
{[][]uint8{{0, 0, 0, 0}, {0, 0, 0, 1}}, Unknown, Unknown, 1, nil},
|
|
{[][]uint8{{0, 0, 0, 0}, {0, 0, 0, 1}}, Unknown, AHash, -1, errors.New("Image hashes's kind should be identical")},
|
|
} {
|
|
data1 := tt.datas[0]
|
|
data2 := tt.datas[1]
|
|
hash1 := NewImageHash(0, tt.hash1)
|
|
hash2 := NewImageHash(0, tt.hash2)
|
|
|
|
for i := 0; i < len(data1); i++ {
|
|
if data1[i] == 1 {
|
|
hash1.Set(i)
|
|
}
|
|
}
|
|
|
|
for i := 0; i < len(data2); i++ {
|
|
if data2[i] == 1 {
|
|
hash2.Set(i)
|
|
}
|
|
}
|
|
|
|
dis, err := hash1.Distance(hash2)
|
|
if dis != tt.distance {
|
|
t.Errorf("Distance between %v and %v expected as %d but got %d", data1, data2, tt.distance, dis)
|
|
}
|
|
if err != nil && err.Error() != tt.err.Error() {
|
|
t.Errorf("Expected err %s, actual %s", tt.err, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSerialization(t *testing.T) {
|
|
checkErr := func(err error) {
|
|
if err != nil {
|
|
t.Errorf("%v", err)
|
|
}
|
|
}
|
|
|
|
methods := []func(img image.Image) (*ImageHash, error){
|
|
AverageHash, PerceptionHash, DifferenceHash,
|
|
}
|
|
examples := []string{
|
|
"_examples/sample1.jpg", "_examples/sample2.jpg", "_examples/sample3.jpg", "_examples/sample4.jpg",
|
|
}
|
|
|
|
for _, ex := range examples {
|
|
file, err := os.Open(ex)
|
|
checkErr(err)
|
|
|
|
defer file.Close()
|
|
|
|
img, _, err := image.Decode(file)
|
|
checkErr(err)
|
|
|
|
for _, method := range methods {
|
|
methodStr := runtime.FuncForPC(reflect.ValueOf(method).Pointer()).Name()
|
|
|
|
hash, err := method(img)
|
|
checkErr(err)
|
|
|
|
hex := hash.ToString()
|
|
// len(kind) == 1, len(":") == 1, len(hash) == 16
|
|
if len(hex) != 18 {
|
|
t.Errorf("Got invalid hex string '%v'; %v of '%v'", hex, methodStr, ex)
|
|
}
|
|
|
|
reHash, err := ImageHashFromString(hex)
|
|
checkErr(err)
|
|
|
|
distance, err := hash.Distance(reHash)
|
|
checkErr(err)
|
|
|
|
if distance != 0 {
|
|
t.Errorf("Original and unserialized objects should be identical, got distance=%v; %v of '%v'", distance, methodStr, ex)
|
|
}
|
|
}
|
|
|
|
// test for ExtIExtImageHash
|
|
hashSizeList := []int{8, 16}
|
|
for _, hashSize := range hashSizeList {
|
|
hash, err := PerceptionHashExtend(img, hashSize)
|
|
checkErr(err)
|
|
|
|
hex := hash.ToString()
|
|
// len(kind) == 1, len(":") == 1
|
|
if len(hex) != hashSize*hashSize/4+2 {
|
|
t.Errorf("Got invalid hex string '%v'; %v of '%v'", hex, "PerceptionHashExtend", ex)
|
|
}
|
|
|
|
reHash, err := ExtImageHashFromString(hex)
|
|
checkErr(err)
|
|
|
|
distance, err := hash.Distance(reHash)
|
|
checkErr(err)
|
|
|
|
if distance != 0 {
|
|
t.Errorf("Original and unserialized objects should be identical, got distance=%v; %v of '%v'", distance, "PerceptionHashExtend", ex)
|
|
}
|
|
}
|
|
}
|
|
}
|