Add tests of the built package

datastax · Nov 11, 2023 · 7005e67 · 7005e67
1 parent 06e4c24
commit 7005e67
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -44,3 +44,14 @@ jobs:
 #          export ASTRA_DB_TOKEN="${{ secrets.E2E_TESTS_ASTRA_DB_TOKEN }}"
 #          export OPEN_AI_KEY="${{ secrets.E2E_TESTS_OPEN_AI_KEY }}"
 #          ./dev/run-e2e-tests.sh
+
+        - name: Build the distribution package
+          run: |
+            poetry build
+
+        - name: Test the distribution package
+          run: |
+            python -m venv /tmp/venv
+            source /tmp/venv/bin/activate
+            pip install dist/ragstack_ai-*.whl
+            pytest tests/unit-tests
diff --git a/tests/unit-tests/test_ragstack.py b/tests/unit-tests/test_ragstack.py
@@ -1,3 +1,5 @@
 
 def test_import():
     from langchain.vectorstores import Cassandra # noqa
+    import langserve # noqa
+    import langsmith # noqa