def test_blobs_setter_dma(): blobs = np.random.random((100, 10, 10)) da = DocumentArray([Document() for _ in range(100)]) da.blobs = blobs np.testing.assert_almost_equal(da.blobs, blobs) for x, doc in zip(blobs, da): np.testing.assert_almost_equal(x, doc.blob)
def test_blobs_wrong_len(): da = DocumentArray([Document() for _ in range(100)]) blobs = np.ones((2, 10, 10)) with pytest.raises(ValueError, match='the number of rows in the'): da.blobs = blobs