def test_borrow(): vecs = VectorStore(128) good = numpy.ndarray(shape=(vecs.nr_dim,), dtype='float32') vecs.borrow(good) bad = numpy.ndarray(shape=(vecs.nr_dim+1,), dtype='float32') with pytest.raises(AssertionError) as excinfo: vecs.borrow(bad)