def runtest(self):
from asdf import AsdfFile, block, util
from asdf.tests import helpers
name, version = parse_schema_filename(self.filename)
if should_skip(name, version):
return
standard_version = self._find_standard_version(name, version)
# Make sure that the examples in the schema files (and thus the
# ASDF standard document) are valid.
buff = helpers.yaml_to_asdf('example: ' + self.example.strip(),
standard_version=standard_version)
ff = AsdfFile(
uri=util.filepath_to_url(os.path.abspath(self.filename)),
ignore_unrecognized_tag=self.ignore_unrecognized_tag,
ignore_version_mismatch=self.ignore_version_mismatch,
)
# Fake an external file
ff2 = AsdfFile({'data': np.empty((1024 * 1024 * 8), dtype=np.uint8)})
ff._external_asdf_by_uri[util.filepath_to_url(
os.path.abspath(
os.path.join(os.path.dirname(self.filename),
'external.asdf')))] = ff2
# Add some dummy blocks so that the ndarray examples work
for i in range(3):
b = block.Block(np.zeros((1024 * 1024 * 8), dtype=np.uint8))
b._used = True
ff.blocks.add(b)
b._array_storage = "streamed"
try:
with pytest.warns(None) as w:
ff._open_impl(ff, buff, mode='rw')
# Do not tolerate any warnings that occur during schema validation
assert len(w) == 0, helpers.display_warnings(w)
except Exception:
print("From file:", self.filename)
raise
# Just test we can write it out. A roundtrip test
# wouldn't always yield the correct result, so those have
# to be covered by "real" unit tests.
if b'external.asdf' not in buff.getvalue():
buff = io.BytesIO()
ff.write_to(buff)
def test_invalid_array_storage():
my_array = np.random.rand(8, 8)
tree = {'my_array': my_array}
ff = asdf.AsdfFile(tree)
with pytest.raises(ValueError):
ff.set_array_storage(my_array, 'foo')
b = block.Block()
b._array_storage = 'foo'
with pytest.raises(ValueError):
ff.blocks.add(b)
with pytest.raises(ValueError):
ff.blocks.remove(b)
