Exemplo n.º 1
0
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)
Exemplo n.º 2
0
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