Spaces:
Running
Running
// _ _ | |
// __ _____ __ ___ ___ __ _| |_ ___ | |
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \ | |
// \ V V / __/ (_| |\ V /| | (_| | || __/ | |
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___| | |
// | |
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved. | |
// | |
// CONTACT: [email protected] | |
// | |
package additional | |
import ( | |
"testing" | |
"github.com/stretchr/testify/assert" | |
"github.com/tailor-inc/graphql" | |
) | |
func TestFeatureProjectionField(t *testing.T) { | |
t.Run("should generate featureProjection argument properly", func(t *testing.T) { | |
// given | |
classname := "Class" | |
// when | |
featureProjection := additionalFeatureProjectionField(classname) | |
// then | |
// the built graphQL field needs to support this structure: | |
// Args: { | |
// algorithm: "a", | |
// dimensions: 1, | |
// learningRate: 2, | |
// iterations: 3, | |
// perplexity: 4 | |
// } | |
// Type: { | |
// vector: [0, 1] | |
// } | |
assert.NotNil(t, featureProjection) | |
assert.Equal(t, "ClassAdditionalFeatureProjection", featureProjection.Type.Name()) | |
assert.NotNil(t, featureProjection.Args) | |
assert.Equal(t, 5, len(featureProjection.Args)) | |
assert.NotNil(t, featureProjection.Args["algorithm"]) | |
assert.NotNil(t, featureProjection.Args["dimensions"]) | |
assert.NotNil(t, featureProjection.Args["learningRate"]) | |
assert.NotNil(t, featureProjection.Args["iterations"]) | |
assert.NotNil(t, featureProjection.Args["perplexity"]) | |
featureProjectionObject, featureProjectionObjectOK := featureProjection.Type.(*graphql.Object) | |
assert.True(t, featureProjectionObjectOK) | |
assert.Equal(t, 1, len(featureProjectionObject.Fields())) | |
assert.NotNil(t, featureProjectionObject.Fields()["vector"]) | |
}) | |
} | |