github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 15:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make Integration Tests "work" again (#106)

Browse Source 
This fixes Issue #104 

The tests for HF Embeddings is skipped because of the segfault issue
mentioned there. Perhaps, a new issue should be created for that?
This commit is contained in:
Delip Rao 2022-11-09 16:26:58 -05:00 committed by GitHub
parent abe4fc04fa
commit 95dd2f140e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/integration_tests/embeddings/test_huggingface.py
View File

@ -1,7 +1,10 @@
"""Test huggingface embeddings.""" """Test huggingface embeddings."""
import unittest
from langchain.embeddings.huggingface import HuggingFaceEmbeddings from langchain.embeddings.huggingface import HuggingFaceEmbeddings
@unittest.skip("This test causes a segfault.")
def test_huggingface_embedding_documents() -> None: def test_huggingface_embedding_documents() -> None:
"""Test huggingface embeddings.""" """Test huggingface embeddings."""
documents = ["foo bar"] documents = ["foo bar"]
@ -11,6 +14,7 @@ def test_huggingface_embedding_documents() -> None:
assert len(output[0]) == 768 assert len(output[0]) == 768
@unittest.skip("This test causes a segfault.")
def test_huggingface_embedding_query() -> None: def test_huggingface_embedding_query() -> None:
"""Test huggingface embeddings.""" """Test huggingface embeddings."""
document = "foo bar" document = "foo bar"

2
tests/integration_tests/test_faiss.py
View File

@ -6,7 +6,7 @@ import pytest
from langchain.docstore.document import Document from langchain.docstore.document import Document
from langchain.docstore.in_memory import InMemoryDocstore from langchain.docstore.in_memory import InMemoryDocstore
from langchain.embeddings.base import Embeddings from langchain.embeddings.base import Embeddings
from langchain.faiss import FAISS from langchain.vectorstores.faiss import FAISS
class FakeEmbeddings(Embeddings): class FakeEmbeddings(Embeddings):