113 lines
2.1 KiB
Go
113 lines
2.1 KiB
Go
package trustGraph
|
|
|
|
import (
|
|
"math"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestBasic(t *testing.T) {
|
|
g := NewGroup()
|
|
g.Add(1, 2, 1)
|
|
g.Add(1, 3, .5)
|
|
g.Add(2, 1, 1)
|
|
g.Add(2, 3, .5)
|
|
g.Add(3, 1, 1)
|
|
g.Add(3, 2, 1)
|
|
|
|
g.InitialTrust(1, 1)
|
|
|
|
out := g.Compute()
|
|
|
|
if out[1] < 0.975 {
|
|
t.Error("Trust in node 1 should be closer to 1.00")
|
|
}
|
|
|
|
if out[2] < 0.93 {
|
|
t.Error("Trust in node 2 should be closer to 1.00")
|
|
}
|
|
if out[3] < 0.4 || out[3] > 0.6 {
|
|
t.Error("Trust in node 3 should be closer to 0.50")
|
|
}
|
|
}
|
|
|
|
func TestRand(t *testing.T) {
|
|
peers := 200
|
|
rand.Seed(time.Now().UTC().UnixNano())
|
|
g := NewGroup()
|
|
|
|
//randomly set actual trust values for peers
|
|
actualTrust := make([]float32, peers)
|
|
for i := 0; i < peers; i++ {
|
|
actualTrust[i] = rand.Float32()
|
|
}
|
|
|
|
// peer0 is set to and granted 100% trust
|
|
actualTrust[0] = 1
|
|
g.InitialTrust(0, 1)
|
|
|
|
// set 30% of trust values to +/- 10% of actual trust
|
|
for i := 0; i < peers; i++ {
|
|
for j := 0; j < peers; j++ {
|
|
if rand.Float32() > .7 {
|
|
g.Add(i, j, randNorm(actualTrust[j]))
|
|
}
|
|
}
|
|
}
|
|
|
|
// compute trust
|
|
out := g.Compute()
|
|
|
|
// find RMS error
|
|
e := float32(0)
|
|
for i := 0; i < peers; i++ {
|
|
x := actualTrust[i] - out[i]
|
|
e += x * x
|
|
}
|
|
e = float32(math.Sqrt(float64(e / float32(peers))))
|
|
|
|
if e > .2 {
|
|
t.Error("RMS Error should be less than 20% for a 30% full trust grid of 200 nodes")
|
|
}
|
|
}
|
|
|
|
// randNorm takes a float and returns a value within +/- 10%,
|
|
// without going over 1
|
|
func randNorm(x float32) float32 {
|
|
r := rand.Float32()*.2 + .9
|
|
x *= r
|
|
if x > 1 {
|
|
return 1
|
|
}
|
|
return x
|
|
}
|
|
|
|
func TestRangeError(t *testing.T) {
|
|
g := NewGroup()
|
|
|
|
err := g.Add(1, 2, 1.1)
|
|
if err.Error() != "Trust amount cannot be greater than 1" {
|
|
t.Error("Expected error")
|
|
}
|
|
|
|
err = g.Add(1, 2, -1)
|
|
if err.Error() != "Trust amount cannot be less than 0" {
|
|
t.Error("Expected error less than 0 error")
|
|
}
|
|
|
|
err = g.Add(1, 2, 1)
|
|
if err != nil {
|
|
t.Error("Did not expected error")
|
|
}
|
|
|
|
err = g.Add(1, 2, 0)
|
|
if err != nil {
|
|
t.Error("Did not expected error")
|
|
}
|
|
|
|
err = g.Add(1, 2, 0.5)
|
|
if err != nil {
|
|
t.Error("Did not expected error")
|
|
}
|
|
}
|