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)