def test_spec_version_select():
spec = AsdfSpec('>=1.1.0')
versions = [AsdfVersion(x) for x in ['1.0.0', '1.0.9', '1.1.0', '1.2.0']]
assert spec.select(versions) == '1.2.0'
assert spec.select(versions[:-1]) == '1.1.0'
assert spec.select(versions[:-2]) == None
def test_spec_version_match():
spec = AsdfSpec('>=1.1.0')
assert spec.match(AsdfVersion('1.1.0'))
assert spec.match(AsdfVersion('1.2.0'))
assert not spec.match(AsdfVersion('1.0.0'))
assert not spec.match(AsdfVersion('1.0.9'))
def test_spec_tuple_match():
spec = AsdfSpec('>=1.1.0')
assert spec.match((1,1,0))
assert spec.match((1,2,0))
assert not spec.match((1,0,0))
assert not spec.match((1,0,9))
def test_spec_tuple_select():
spec = AsdfSpec('>=1.1.0')
versions = [(1,0,0), (1,0,9), (1,1,0), (1,2,0)]
assert spec.select(versions) == '1.2.0'
assert spec.select(versions[:-1]) == '1.1.0'
assert spec.select(versions[:-2]) == None
def test_spec_string_select():
spec = AsdfSpec('>=1.1.0')
versions = ['1.0.0', '1.0.9', '1.1.0', '1.2.0']
assert spec.select(versions) == '1.2.0'
assert spec.select(versions[:-1]) == '1.1.0'
assert spec.select(versions[:-2]) == None
def test_spec_string_match():
spec = AsdfSpec('>=1.1.0')
assert spec.match('1.1.0')
assert spec.match('1.2.0')
assert not spec.match('1.0.0')
assert not spec.match('1.0.9')
def test_spec_string_match():
spec = AsdfSpec('>=1.1.0')
assert spec.match('1.1.0')
assert spec.match('1.2.0')
assert not spec.match('1.0.0')
assert not spec.match('1.0.9')
def test_spec_version_match():
spec = AsdfSpec('>=1.1.0')
assert spec.match(AsdfVersion('1.1.0'))
assert spec.match(AsdfVersion('1.2.0'))
assert not spec.match(AsdfVersion('1.0.0'))
assert not spec.match(AsdfVersion('1.0.9'))
def test_spec_version_select():
spec = AsdfSpec('>=1.1.0')
versions = [AsdfVersion(x) for x in ['1.0.0', '1.0.9', '1.1.0', '1.2.0']]
assert spec.select(versions) == '1.2.0'
assert spec.select(versions[:-1]) == '1.1.0'
assert spec.select(versions[:-2]) == None
def test_spec_tuple_select():
spec = AsdfSpec('>=1.1.0')
versions = [(1,0,0), (1,0,9), (1,1,0), (1,2,0)]
assert spec.select(versions) == '1.2.0'
assert spec.select(versions[:-1]) == '1.1.0'
assert spec.select(versions[:-2]) == None
def test_spec_tuple_match():
spec = AsdfSpec('>=1.1.0')
assert spec.match((1,1,0))
assert spec.match((1,2,0))
assert not spec.match((1,0,0))
assert not spec.match((1,0,9))
def test_spec_string_select():
spec = AsdfSpec('>=1.1.0')
versions = ['1.0.0', '1.0.9', '1.1.0', '1.2.0']
assert spec.select(versions) == '1.2.0'
assert spec.select(versions[:-1]) == '1.1.0'
assert spec.select(versions[:-2]) == None
def test_spec_equal():
"""Make sure that equality means match"""
spec = AsdfSpec('>=1.2.0')
version0 = AsdfVersion('1.1.0')
version1 = AsdfVersion('1.3.0')
assert spec != version0
assert version0 != spec
assert spec == version1
assert version1 == spec
assert spec != '1.1.0'
assert '1.1.0' != spec
assert spec == '1.3.0'
assert '1.3.0' == spec
assert spec != (1, 1, 0)
assert (1, 1, 0) != spec
assert spec == (1, 3, 0)
assert (1, 3, 0) == spec
def test_spec_tuple_filter():
spec = AsdfSpec('>=1.1.0')
versions = [(1,0,0), (1,0,9), (1,1,0), (1,2,0)]
for x,y in zip(spec.filter(versions), ['1.1.0', '1.2.0']):
assert x == y
def test_spec_string_filter():
spec = AsdfSpec('>=1.1.0')
versions = ['1.0.0', '1.0.9', '1.1.0', '1.2.0']
for x,y in zip(spec.filter(versions), ['1.1.0', '1.2.0']):
assert x == y
def test_spec_tuple_filter():
spec = AsdfSpec('>=1.1.0')
versions = [(1,0,0), (1,0,9), (1,1,0), (1,2,0)]
for x,y in zip(spec.filter(versions), ['1.1.0', '1.2.0']):
assert x == y
def test_spec_string_filter():
spec = AsdfSpec('>=1.1.0')
versions = ['1.0.0', '1.0.9', '1.1.0', '1.2.0']
for x,y in zip(spec.filter(versions), ['1.1.0', '1.2.0']):
assert x == y
def test_spec_version_filter():
spec = AsdfSpec('>=1.1.0')
versions = [AsdfVersion(x) for x in ['1.0.0', '1.0.9', '1.1.0', '1.2.0']]
for x,y in zip(spec.filter(versions), ['1.1.0', '1.2.0']):
assert x == y
class TimeType(AstropyAsdfType):
name = 'time/time'
version = '1.1.0'
supported_versions = ['1.0.0', AsdfSpec('>=1.1.0')]
types = ['astropy.time.core.Time']
requires = ['astropy']
@classmethod
def to_tree(cls, node, ctx):
format = node.format
if format == 'byear':
node = time.Time(node, format='byear_str')
elif format == 'jyear':
node = time.Time(node, format='jyear_str')
elif format in ('fits', 'datetime', 'plot_date'):
node = time.Time(node, format='isot')
format = node.format
format = _astropy_format_to_asdf_format.get(format, format)
guessable_format = format in _guessable_formats
if node.scale == 'utc' and guessable_format:
if node.isscalar:
return node.value
else:
return yamlutil.custom_tree_to_tagged_tree(node.value, ctx)
d = {'value': yamlutil.custom_tree_to_tagged_tree(node.value, ctx)}
if not guessable_format:
d['format'] = format
if node.scale != 'utc':
d['scale'] = node.scale
if node.location is not None:
x, y, z = node.location.x, node.location.y, node.location.z
# Preserve backwards compatibility for writing the old schema
# This allows WCS to test backwards compatibility with old frames
# This code does get tested in CI, but we don't run a coverage test
if cls.version == '1.0.0': # pragma: no cover
unit = node.location.unit
d['location'] = {'x': x, 'y': y, 'z': z, 'unit': unit}
else:
d['location'] = {
# It seems like EarthLocations can be represented either in
# terms of Cartesian coordinates or latitude and longitude, so
# we rather arbitrarily choose the former for our representation
'x': yamlutil.custom_tree_to_tagged_tree(x, ctx),
'y': yamlutil.custom_tree_to_tagged_tree(y, ctx),
'z': yamlutil.custom_tree_to_tagged_tree(z, ctx)
}
return d
@classmethod
def from_tree(cls, node, ctx):
if isinstance(node, (str, list, np.ndarray)):
t = time.Time(node)
format = _astropy_format_to_asdf_format.get(t.format, t.format)
if format not in _guessable_formats:
raise ValueError("Invalid time '{0}'".format(node))
return t
value = node['value']
format = node.get('format')
scale = node.get('scale')
location = node.get('location')
if location is not None:
unit = location.get('unit', u.m)
# This ensures that we can read the v.1.0.0 schema and convert it
# to the new EarthLocation object, which expects Quantity components
for comp in ['x', 'y', 'z']:
if not isinstance(location[comp], Quantity):
location[comp] = Quantity(location[comp], unit=unit)
location = EarthLocation.from_geocentric(location['x'],
location['y'],
location['z'])
return time.Time(value, format=format, scale=scale, location=location)
@classmethod
def assert_equal(cls, old, new):
assert old.format == new.format
assert old.scale == new.scale
if isinstance(old.location, EarthLocation):
assert isinstance(new.location, EarthLocation)
_assert_earthlocation_equal(old.location, new.location)
else:
assert old.location == new.location
assert_array_equal(old, new)
def test_spec_version_filter():
spec = AsdfSpec('>=1.1.0')
versions = [AsdfVersion(x) for x in ['1.0.0', '1.0.9', '1.1.0', '1.2.0']]
for x,y in zip(spec.filter(versions), ['1.1.0', '1.2.0']):
assert x == y