From 634066a4bbc31701b3d80fec6feaef5a686b1f48 Mon Sep 17 00:00:00 2001 From: Arjun Sunil Kumar Date: Thu, 15 Feb 2024 13:59:52 -0800 Subject: [PATCH] minor --- utils/moarray/internal.go | 2 +- utils/moarray/internal_test.go | 111 +++++++++++++++++---------------- 2 files changed, 59 insertions(+), 54 deletions(-) diff --git a/utils/moarray/internal.go b/utils/moarray/internal.go index 803faad..9f6a671 100644 --- a/utils/moarray/internal.go +++ b/utils/moarray/internal.go @@ -37,6 +37,6 @@ func NormalizeGonumVectors(vectors []*mat.VecDense) { // NormalizeMoVecf64 is used only in test functions. func NormalizeMoVecf64(vector []float64) []float64 { res := ToGonumVector[float64](vector) - //NormalizeGonumVector(res) + NormalizeGonumVector(res) return ToMoArray[float64](res) } diff --git a/utils/moarray/internal_test.go b/utils/moarray/internal_test.go index 82d9077..f2094b4 100644 --- a/utils/moarray/internal_test.go +++ b/utils/moarray/internal_test.go @@ -14,56 +14,61 @@ package moarray -//func TestNormalizeMoArray(t *testing.T) { -// type args struct { -// vector []float64 -// } -// tests := []struct { -// name string -// args args -// want []float64 -// }{ -// { -// name: "Test1", -// args: args{ -// vector: []float64{1, 2, 3}, -// }, -// want: []float64{0.2672612419124244, 0.5345224838248488, 0.8017837257372732}, -// }, -// { -// name: "Test2", -// args: args{ -// vector: []float64{-1, 2, 3}, -// }, -// want: []float64{-0.2672612419124244, 0.5345224838248488, 0.8017837257372732}, -// }, -// { -// name: "Test3", -// args: args{ -// vector: []float64{0, 0, 0}, -// }, -// want: []float64{0, 0, 0}, -// }, -// { -// name: "Test4", -// args: args{ -// vector: []float64{10, 3.333333333333333, 4, 5}, -// }, -// want: []float64{0.8108108108108107, 0.27027027027027023, 0.3243243243243243, 0.4054054054054054}, -// }, -// { -// name: "Test5", -// args: args{ -// vector: []float64{1, 2, 3.6666666666666665, 4.666666666666666}, -// }, -// want: []float64{0.15767649936829103, 0.31535299873658207, 0.5781471643504005, 0.7358236637186913}, -// }, -// } -// for _, tt := range tests { -// t.Run(tt.name, func(t *testing.T) { -// if got := NormalizeMoVecf64(tt.args.vector); !assertx.InEpsilonF64Slice(tt.want, got) { -// t.Errorf("NormalizeMoVecf64() = %v, want %v", got, tt.want) -// } -// }) -// } -//} +import ( + "github.com/arjunsk/kmeans/utils/assertx" + "testing" +) + +func TestNormalizeMoArray(t *testing.T) { + type args struct { + vector []float64 + } + tests := []struct { + name string + args args + want []float64 + }{ + { + name: "Test1", + args: args{ + vector: []float64{1, 2, 3}, + }, + want: []float64{0.2672612419124244, 0.5345224838248488, 0.8017837257372732}, + }, + { + name: "Test2", + args: args{ + vector: []float64{-1, 2, 3}, + }, + want: []float64{-0.2672612419124244, 0.5345224838248488, 0.8017837257372732}, + }, + { + name: "Test3", + args: args{ + vector: []float64{0, 0, 0}, + }, + want: []float64{0, 0, 0}, + }, + { + name: "Test4", + args: args{ + vector: []float64{10, 3.333333333333333, 4, 5}, + }, + want: []float64{0.8108108108108107, 0.27027027027027023, 0.3243243243243243, 0.4054054054054054}, + }, + { + name: "Test5", + args: args{ + vector: []float64{1, 2, 3.6666666666666665, 4.666666666666666}, + }, + want: []float64{0.15767649936829103, 0.31535299873658207, 0.5781471643504005, 0.7358236637186913}, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + if got := NormalizeMoVecf64(tt.args.vector); !assertx.InEpsilonF64Slice(tt.want, got) { + t.Errorf("NormalizeMoVecf64() = %v, want %v", got, tt.want) + } + }) + } +}