github.com/devwanda/aphelion-staking@v0.33.9/crypto/tmhash/hash_test.go (about) 1 package tmhash_test 2 3 import ( 4 "crypto/sha256" 5 "testing" 6 7 "github.com/stretchr/testify/assert" 8 9 "github.com/devwanda/aphelion-staking/crypto/tmhash" 10 ) 11 12 func TestHash(t *testing.T) { 13 testVector := []byte("abc") 14 hasher := tmhash.New() 15 hasher.Write(testVector) 16 bz := hasher.Sum(nil) 17 18 bz2 := tmhash.Sum(testVector) 19 20 hasher = sha256.New() 21 hasher.Write(testVector) 22 bz3 := hasher.Sum(nil) 23 24 assert.Equal(t, bz, bz2) 25 assert.Equal(t, bz, bz3) 26 } 27 28 func TestHashTruncated(t *testing.T) { 29 testVector := []byte("abc") 30 hasher := tmhash.NewTruncated() 31 hasher.Write(testVector) 32 bz := hasher.Sum(nil) 33 34 bz2 := tmhash.SumTruncated(testVector) 35 36 hasher = sha256.New() 37 hasher.Write(testVector) 38 bz3 := hasher.Sum(nil) 39 bz3 = bz3[:tmhash.TruncatedSize] 40 41 assert.Equal(t, bz, bz2) 42 assert.Equal(t, bz, bz3) 43 }