mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-03 07:04:01 +00:00
24 lines
837 B
Python
24 lines
837 B
Python
import pytest
|
|
|
|
from haystack.dataclasses.sparse_embedding import SparseEmbedding
|
|
|
|
|
|
class TestSparseEmbedding:
|
|
def test_init(self):
|
|
se = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
|
|
assert se.indices == [0, 2, 4]
|
|
assert se.values == [0.1, 0.2, 0.3]
|
|
|
|
def test_init_with_wrong_parameters(self):
|
|
with pytest.raises(ValueError):
|
|
SparseEmbedding(indices=[0, 2], values=[0.1, 0.2, 0.3, 0.4])
|
|
|
|
def test_to_dict(self):
|
|
se = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
|
|
assert se.to_dict() == {"indices": [0, 2, 4], "values": [0.1, 0.2, 0.3]}
|
|
|
|
def test_from_dict(self):
|
|
se = SparseEmbedding.from_dict({"indices": [0, 2, 4], "values": [0.1, 0.2, 0.3]})
|
|
assert se.indices == [0, 2, 4]
|
|
assert se.values == [0.1, 0.2, 0.3]
|