def test_invalid_modes(self):
# Modes 'r+', 'r', and 'w+' cannot work with Aesara, becausei
# the output array may be modified inplace, and that should not
# modify the original file.
path = Variable(Generic())
for mmap_mode in ("r+", "r", "w+", "toto"):
with pytest.raises(ValueError):
load(path, "int32", (False,), mmap_mode)
def test_basic(self):
path = Variable(Generic())
# Not specifying mmap_mode defaults to None, and the data is
# copied into main memory
x = load(path, "int32", (False,))
y = x * 2
fn = function([path], y)
assert (fn(self.filename) == (self.data * 2)).all()
def test1(self):
path = Variable(Generic())
# 'c' means "copy-on-write", which allow the array to be overwritten
# by an inplace Op in the graph, without modifying the underlying
# file.
x = load(path, "int32", (False,), "c")
# x ** 2 has been chosen because it will work inplace.
y = (x ** 2).sum()
fn = function([path], y)
# Call fn() twice, to check that inplace ops do not cause trouble
assert (fn(self.filename) == (self.data ** 2).sum()).all()
assert (fn(self.filename) == (self.data ** 2).sum()).all()
def test_memmap(self):
path = Variable(Generic())
x = load(path, "int32", (False,), mmap_mode="c")
fn = function([path], x)
assert type(fn(self.filename)) == np.core.memmap