def setUp(self):
if self.file_type == 'filename':
fd, path = tempfile.mkstemp()
os.close(fd)
self.file = path
elif self.file_type == 'bytesio':
self.file = six.BytesIO()
else:
assert False
# Create and save a link
child = link.Chain()
with child.init_scope():
child.linear = links.Linear(2, 3)
parent = link.Chain()
with parent.init_scope():
parent.linear = links.Linear(3, 2)
parent.child = child
npz.save_npz(self.file, parent)
self.source = parent
if self.file_type == 'bytesio':
self.file.seek(0)
self.npzfile = numpy.load(self.file)
self.deserializer = npz.NpzDeserializer(self.npzfile, strict=False)
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.temp_file_path = path
with open(path, 'wb') as f:
numpy.savez(f, **{'x': numpy.asarray(10)})
self.npzfile = numpy.load(path)
self.deserializer = npz.NpzDeserializer(self.npzfile, strict=False)
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.temp_file_path = path
with open(path, 'wb') as f:
numpy.savez(
f, **{'x': numpy.asarray(10), 'yy': numpy.empty((2, 3))})
self.npzfile = numpy.load(path)
self.deserializer = npz.NpzDeserializer(
self.npzfile, ignore_names=self.ignore_names)
def setUp(self):
self.data = numpy.random.uniform(-1, 1, (2, 3)).astype(numpy.float32)
fd, path = tempfile.mkstemp()
os.close(fd)
self.temp_file_path = path
with open(path, 'wb') as f:
savez = numpy.savez_compressed if self.compress else numpy.savez
savez(f, **{'x/': None, 'y': self.data, 'z': numpy.asarray(10)})
self.npzfile = numpy.load(path)
self.deserializer = npz.NpzDeserializer(self.npzfile)
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.temp_file_path = path
child = link.Chain(linear=links.Linear(2, 3))
parent = link.Chain(linear=links.Linear(3, 2), child=child)
npz.save_npz(self.temp_file_path, parent)
self.source = parent
self.npzfile = numpy.load(path)
self.deserializer = npz.NpzDeserializer(self.npzfile, strict=False)
def setUp(self):
fd, path = tempfile.mkstemp()
os.close(fd)
self.temp_file_path = path
child = link.Chain()
with child.init_scope():
child.linear2 = links.Linear(2, 3)
parent = link.Chain()
with parent.init_scope():
parent.linear = links.Linear(3, 2)
parent.child = child
npz.save_npz(self.temp_file_path, parent)
self.source = parent
self.npzfile = numpy.load(path)
self.deserializer = npz.NpzDeserializer(self.npzfile,
ignore_names=self.ignore_names)
def setUp(self):
self.data = numpy.random.uniform(-1, 1, (2, 3)).astype(numpy.float32)
fd, path = tempfile.mkstemp()
os.close(fd)
self.temp_file_path = path
with open(path, 'wb') as f:
savez = numpy.savez_compressed if self.compress else numpy.savez
savez(
f, **{
'x/': None,
'y': self.data,
'z': numpy.asarray(10),
'zf32': numpy.array(-2**60, dtype=numpy.float32),
'zi64': numpy.array(-2**60, dtype=numpy.int64),
'w': None
})
try:
self.npzfile = numpy.load(path, allow_pickle=True)
except TypeError:
self.npzfile = numpy.load(path)
self.deserializer = npz.NpzDeserializer(self.npzfile)
