12345678910111213141516171819202122232425262728293031323334353637 |
- // Copyright 2022 gorse Project Authors
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package search
- import (
- "testing"
- "github.com/stretchr/testify/assert"
- )
- func TestHNSWConfig(t *testing.T) {
- hnsw := &HNSW{}
- SetHNSWNumJobs(123)(hnsw)
- assert.Equal(t, 123, hnsw.numJobs)
- SetMaxConnection(234)(hnsw)
- assert.Equal(t, 234, hnsw.maxConnection)
- SetEFConstruction(345)(hnsw)
- assert.Equal(t, 345, hnsw.efConstruction)
- SetEF(456)(hnsw)
- assert.Equal(t, 456, hnsw.ef)
- }
|