refactor: move tests to one file

KevKibe · May 14, 2024 · 71bf011 · 71bf011
1 parent 4eb40b2
commit 71bf011
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 61 deletions.
diff --git a/.github/workflows/test_google_pinecone_indexer.yaml b/.github/workflows/test_google_pinecone_indexer.yaml
diff --git a/.github/workflows/test_openai_pinecone_indexer.yaml b/.github/workflows/test_openai_pinecone_indexer.yaml
diff --git a/...rkflows/test_cohere_pinecone_indexer.yaml → .github/workflows/tests.yaml b/...rkflows/test_cohere_pinecone_indexer.yaml → .github/workflows/tests.yaml
@@ -20,10 +20,30 @@ jobs:
           pip install -r requirements.txt --no-cache-dir
           pip install pytest einops lion-pytorch
 
-      - name: Test with pytest
+      - name: Test CoherePineconeIndexer
+        id: cohere_test
         env:
             PINECONE_API_KEY: ${{ secrets.PINECONE_API_KEY }}
             COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }}
         run: |
           pytest src/tests/cohereindex_test.py
+
+
+      - name: Test OpenaiPineconeIndexer
+        id: openai_test
+        env:
+            PINECONE_API_KEY: ${{ secrets.PINECONE_API_KEY }}
+            OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+        run: |
+          pytest src/tests/openaiindex_test.py
+
+
+      - name: Test GooglePineconeIndexer
+        id: google_test
+        env:
+            PINECONE_API_KEY: ${{ secrets.PINECONE_API_KEY }}
+            GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
+        run: |
+          pytest src/tests/googleindex_test.py
+      
 
diff --git a/README.md b/README.md
@@ -140,7 +140,7 @@ cd src
 
 ```bash
 # Using OpenAI 
-python -m  _openai.create_index --pinecone_api_key "your_pinecone_api_key" --index_name "your_index_name" -
+python -m  utils.create_index --pinecone_api_key "your_pinecone_api_key" --index_name "your_index_name" 
 
 ```
 
@@ -164,7 +164,7 @@ python -m _cohere.index_documents   --pinecone_api_key "your_pinecone_api_key" -
 
 ```bash
 # Using OpenAI 
-python -m  _openai.delete_index --pinecone_api_key "your_pinecone_api_key" --index_name "your_index_name" 
+python -m  utils.delete_index --pinecone_api_key "your_pinecone_api_key" --index_name "your_index_name" 
 
 ```