예제 #1
0
def test_borrow():
    vecs = Vectors(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)
예제 #2
0
def test_borrow():
    vecs = Vectors(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)