Skip to content

Commit

Permalink
fix: Update to include rag unit tests
Browse files Browse the repository at this point in the history
Signed-off-by: ishaansehgal99 <ishaanforthewin@gmail.com>
  • Loading branch information
ishaansehgal99 committed Oct 3, 2024
1 parent 6e7b827 commit aaaa21b
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 3 deletions.
4 changes: 3 additions & 1 deletion .github/workflows/tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,10 @@ jobs:
- name: Run unit tests & Generate coverage
run: |
make unit-test
make rag-service-test
make tuning-metrics-server-test
- name: Run inference api unit tests
- name: Run inference api e2e tests
run: |
make inference-api-e2e
Expand Down
14 changes: 12 additions & 2 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -96,13 +96,23 @@ unit-test: ## Run unit tests.
-race -coverprofile=coverage.txt -covermode=atomic
go tool cover -func=coverage.txt

.PHONY: rag-service-test
rag-service-test:
pip install -r presets/rag_service/requirements.txt
pytest -o log_cli=true -o log_cli_level=INFO presets/rag_service/tests

.PHONY: tuning-metrics-server-test
tuning-metrics-server-test:
pytest -o log_cli=true -o log_cli_level=INFO presets/tuning/text-generation/metrics

## --------------------------------------
## E2E tests
## --------------------------------------

inference-api-e2e:
.PHONY: inference-api-e2e
inference-api-e2e:
pip install -r presets/inference/text-generation/requirements.txt
pytest -o log_cli=true -o log_cli_level=INFO .
pytest -o log_cli=true -o log_cli_level=INFO presets/inference/text-generation/tests

# Ginkgo configurations
GINKGO_FOCUS ?=
Expand Down

0 comments on commit aaaa21b

Please sign in to comment.