wehub-resource-sync
c56bef871b
Sync docs with Docusaurus / sync (push) Waiting to run
Tests / Check if changed (push) Waiting to run
Tests / format (push) Blocked by required conditions
Tests / check-imports (push) Blocked by required conditions
Tests / Unit / macos-latest (push) Blocked by required conditions
Tests / Unit / ubuntu-latest (push) Blocked by required conditions
Tests / Unit / windows-latest (push) Blocked by required conditions
Tests / mypy (push) Blocked by required conditions
Tests / Integration / ubuntu-latest (push) Blocked by required conditions
Tests / Integration / macos-latest (push) Blocked by required conditions
Tests / Integration / windows-latest (push) Blocked by required conditions
Tests / notify-slack-on-failure (push) Blocked by required conditions
Tests / Mark tests as completed (push) Blocked by required conditions
Docker image release / Build base image (push) Waiting to run
CodeQL / Analyze (python) (push) Has been cancelled
Update Platform Components Table / update (push) Has been cancelled
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import warnings
|
|
|
|
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]
|
|
|
|
def test_eq(self):
|
|
se1 = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
|
|
se2 = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
|
|
assert se1 == se2
|
|
|
|
se3 = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.4])
|
|
assert se1 != se3
|
|
|
|
def test_no_warning_on_init(self):
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("error", Warning)
|
|
SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
|
|
|
|
def test_warn_on_inplace_mutation(self):
|
|
se = SparseEmbedding(indices=[0, 2, 4], values=[0.1, 0.2, 0.3])
|
|
with pytest.warns(Warning, match="dataclasses.replace"):
|
|
se.indices = [1, 3, 5]
|