(self)
| 85 | assert doc.vector("keys") == {1: 1.0, 2: 2.0, 3: 3.0} |
| 86 | |
| 87 | def test_init_normalizes_numpy_vectors(self): |
| 88 | import numpy as np |
| 89 | |
| 90 | doc = Doc(id="1", vectors={"dense": np.array([1, 2, 3])}) |
| 91 | |
| 92 | assert doc.vector("dense") == [1, 2, 3] |
| 93 | assert json.loads(repr(doc))["vectors"]["dense"] == [1, 2, 3] |
| 94 | |
| 95 | |
| 96 | # ---------------------------- |