Skip to content

Commit

Permalink
feat: added add route tfidf test
Browse files Browse the repository at this point in the history
  • Loading branch information
digriffiths committed Jan 8, 2024
1 parent 138d540 commit 3698b07
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions tests/unit/test_hybrid_layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -165,5 +160,18 @@ def test_failover_score_threshold(self, base_encoder, bm25_encoder):
)
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 comments on commit 3698b07

Please sign in to comment.