Skip to content
Snippets Groups Projects
Commit 3698b07f authored by “Daniel Griffiths”'s avatar “Daniel Griffiths”
Browse files

feat: added add route tfidf test

parent 138d540b
No related branches found
No related tags found
No related merge requests found
......@@ -30,11 +30,6 @@ def base_encoder(mocker):
return mock_base_encoder
# @pytest.fixture
# def base_encoder():
# return BaseEncoder(name="test-encoder")
@pytest.fixture
def cohere_encoder(mocker):
mocker.patch.object(CohereEncoder, "__call__", side_effect=mock_encoder_call)
......@@ -165,5 +160,18 @@ class TestHybridRouteLayer:
)
assert route_layer.score_threshold == 0.82
def test_add_route_tfidf(self, cohere_encoder, tfidf_encoder, routes):
hybrid_route_layer = HybridRouteLayer(
dense_encoder=cohere_encoder,
sparse_encoder=tfidf_encoder,
routes=routes[:-1],
)
hybrid_route_layer.add(routes[-1])
all_utterances = [
utterance for route in routes for utterance in route.utterances
]
assert hybrid_route_layer.sparse_index is not None
assert len(hybrid_route_layer.sparse_index) == len(all_utterances)
# Add more tests for edge cases and error handling as needed.
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment