def test_blobs_wrong_len(tmpdir): dam = DocumentArrayMemmap(tmpdir) dam.extend([Document() for x in range(100)]) blobs = np.ones((2, 10, 10)) with pytest.raises(ValueError): dam.blobs = blobs
def test_blobs_setter_dam(tmpdir): blobs = np.random.random((100, 10, 10)) dam = DocumentArrayMemmap(tmpdir) dam.extend([Document() for _ in blobs]) dam.blobs = blobs np.testing.assert_almost_equal(dam.blobs, blobs) for x, doc in zip(blobs, dam): np.testing.assert_almost_equal(x, doc.blob)