Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Write unit tests for vector_db.py #4088

Closed
wants to merge 1 commit into from

Conversation

sweep-nightly[bot]
Copy link
Contributor

@sweep-nightly sweep-nightly bot commented Jun 21, 2024

Purpose

This pull request adds unit tests for the vector_db.py module to ensure the reliability and correctness of the vector database implementation.

Description

The changes in this pull request include:

  1. Creation of a new file test_vector_db.py in the same directory as vector_db.py to house the unit tests.
  2. Setup of test fixtures for mocking the Redis and OpenAI clients, as well as providing sample test data.
  3. Implementation of the following test cases:
    • Test for embed_text_array function
    • Test for openai_call_embedding function, including error handling
    • Test for apply_adjustment_score function with parametrized inputs
    • Test for get_pointwise_reranked_snippet_scores function

Summary

  • Created test_vector_db.py file to house unit tests for vector_db.py
  • Implemented test fixtures for mocking dependencies and providing sample test data
  • Added unit tests for the following functions:
    • embed_text_array
    • openai_call_embedding
    • apply_adjustment_score
    • get_pointwise_reranked_snippet_scores

Suggested changes from Sweep Chat by @kevinlu1248. Continue chatting at http://localhost:3000/c/73163217-5887-4be9-85b4-cbf80a72ef33.

Copy link

vercel bot commented Jun 21, 2024

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name Status Preview Comments Updated (UTC)
sweep-chat 🔄 Building (Inspect) Visit Preview 💬 Add feedback Jun 21, 2024 9:31pm
sweep-chat-demo ✅ Ready (Inspect) Visit Preview 💬 Add feedback Jun 21, 2024 9:31pm
sweep-docs ✅ Ready (Inspect) Visit Preview 💬 Add feedback Jun 21, 2024 9:31pm
sweep-pulley 🔄 Building (Inspect) Visit Preview 💬 Add feedback Jun 21, 2024 9:31pm

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant