Spaces:
Running
Running
File size: 2,827 Bytes
b110593 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package modcentroid
import (
"context"
"fmt"
"testing"
"github.com/stretchr/testify/assert"
"github.com/weaviate/weaviate/entities/models"
"github.com/weaviate/weaviate/entities/moduletools"
"github.com/weaviate/weaviate/modules/ref2vec-centroid/config"
)
func TestConfigDefaults(t *testing.T) {
def := New().ClassConfigDefaults()
cfg := config.New(fakeClassConfig(def))
assert.Equal(t, config.MethodDefault, cfg.CalculationMethod())
}
func TestConfigValidator(t *testing.T) {
class := &models.Class{Class: "CentroidClass"}
tests := []struct {
name string
class *models.Class
classConfig moduletools.ClassConfig
expectedErr error
}{
{
name: "valid config",
class: class,
classConfig: fakeClassConfig{
"referenceProperties": []interface{}{"someRef"},
},
},
{
name: "invalid config - required fields omitted",
class: class,
classConfig: fakeClassConfig{},
expectedErr: fmt.Errorf("validate %q: invalid config: must have at least "+
"one value in the \"referenceProperties\" field",
class.Class),
},
{
name: "invalid config - wrong type for referenceProperties",
class: class,
classConfig: fakeClassConfig{
"referenceProperties": "someRef",
},
expectedErr: fmt.Errorf("validate %q: invalid config: expected array for "+
"field \"referenceProperties\", got string",
class.Class),
},
{
name: "invalid config - empty referenceProperties slice",
class: class,
classConfig: fakeClassConfig{
"referenceProperties": []interface{}{},
},
expectedErr: fmt.Errorf("validate %q: invalid config: must have at least "+
"one value in the \"referenceProperties\" field",
class.Class),
},
{
name: "invalid config - non-string value in referenceProperties array",
class: class,
classConfig: fakeClassConfig{
"referenceProperties": []interface{}{"someRef", 123},
},
expectedErr: fmt.Errorf("validate %q: invalid config: expected \"referenceProperties\" "+
"to contain strings, found int: [someRef 123]",
class.Class),
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
mod := New()
err := mod.ValidateClass(context.Background(), test.class, test.classConfig)
if test.expectedErr != nil {
assert.EqualError(t, err, test.expectedErr.Error())
} else {
assert.Nil(t, err)
}
})
}
}
|