my-project-public

repository

loading code, commits, and activity

repositories

loading repo index

my-project-public — gitlawb

#1	import pytest
#2	from chromadb.api.types import Documents, Embeddings
#3
#4	from embedchain.config.embedder.base import BaseEmbedderConfig
#5	from embedchain.embedder.base import BaseEmbedder
#6
#7
#8	@pytest.fixture
#9	def base_embedder():
#10	return BaseEmbedder()
#11
#12
#13	def test_initialization(base_embedder):
#14	assert isinstance(base_embedder.config, BaseEmbedderConfig)
#15	# not initialized
#16	assert not hasattr(base_embedder, "embedding_fn")
#17	assert not hasattr(base_embedder, "vector_dimension")
#18
#19
#20	def test_set_embedding_fn(base_embedder):
#21	def embedding_function(texts: Documents) -> Embeddings:
#22	return [f"Embedding for {text}" for text in texts]
#23
#24	base_embedder.set_embedding_fn(embedding_function)
#25	assert hasattr(base_embedder, "embedding_fn")
#26	assert callable(base_embedder.embedding_fn)
#27	embeddings = base_embedder.embedding_fn(["text1", "text2"])
#28	assert embeddings == ["Embedding for text1", "Embedding for text2"]
#29
#30
#31	def test_set_embedding_fn_when_not_a_function(base_embedder):
#32	with pytest.raises(ValueError):
#33	base_embedder.set_embedding_fn(None)
#34
#35
#36	def test_set_vector_dimension(base_embedder):
#37	base_embedder.set_vector_dimension(256)
#38	assert hasattr(base_embedder, "vector_dimension")
#39	assert base_embedder.vector_dimension == 256
#40
#41
#42	def test_set_vector_dimension_type_error(base_embedder):
#43	with pytest.raises(TypeError):
#44	base_embedder.set_vector_dimension(None)
#45
#46
#47	def test_embedder_with_config():
#48	embedder = BaseEmbedder(BaseEmbedderConfig())
#49	assert isinstance(embedder.config, BaseEmbedderConfig)
#50

z6Mkq5mY3JWtxoxUobWcfNHm7AkRubgSWEZTkBVqZXJviFZ5/my-project-public