def test_embeddings_wrong_len(tmpdir):
dam = DocumentArrayMemmap(tmpdir)
dam.extend([Document() for x in range(100)])
embeddings = np.ones((2, 10, 10))
with pytest.raises(ValueError):
dam.embeddings = embeddings
def test_embeddings_setter_dam(tmpdir):
emb = np.random.random((100, 128))
dam = DocumentArrayMemmap(tmpdir)
dam.extend([Document() for _ in range(100)])
dam.embeddings = emb
np.testing.assert_almost_equal(dam.embeddings, emb)
for x, doc in zip(emb, dam):
np.testing.assert_almost_equal(x, doc.embedding)
