def setup_readwrite(self):
"""Setup & teardown for read/write tests."""
# register
io_registry.register_reader("json", Cosmology, read_json)
io_registry.register_writer("json", Cosmology, write_json)
io_registry.register_identifier("json", Cosmology, json_identify)
yield # run all tests in class
# unregister
io_registry.unregister_reader("json", Cosmology)
io_registry.unregister_writer("json", Cosmology)
io_registry.unregister_identifier("json", Cosmology)
def write_obslog(table, filename, delimiter=' '):
"""Write an observing log table to an ASCII file.
Args:
table (:class:`astropy.table.Table`): Observing log table.
filename (str): Name of the output ASCII file.
delimiter (str): A character as the delimiter between columns.
"""
registry.register_writer('obslog', Table, _write_obslog)
table.write(filename, format='obslog', delimiter=delimiter)
registry.unregister_writer('obslog', Table)
def teardown():
"""Clean up module after tests."""
yield # to let all tests within the scope run
if ASTROPY_GE_5:
from astropy.cosmology import Cosmology
from astropy.io import registry as io_registry
io_registry.unregister_reader("myformat", Cosmology)
io_registry.unregister_writer("myformat", Cosmology)
io_registry.unregister_identifier("myformat", Cosmology)
io_registry.unregister_reader("mypackage", Cosmology)
io_registry.unregister_writer("mypackage", Cosmology)
io_registry.unregister_identifier("mypackage", Cosmology)
def test_register_writer():
io_registry.register_writer('test1', TestData, empty_writer)
io_registry.register_writer('test2', TestData, empty_writer)
assert io_registry.get_writer('test1', TestData) == empty_writer
assert io_registry.get_writer('test2', TestData) == empty_writer
io_registry.unregister_writer('test1', TestData)
with pytest.raises(io_registry.IORegistryError):
io_registry.get_writer('test1', TestData)
assert io_registry.get_writer('test2', TestData) == empty_writer
io_registry.unregister_writer('test2', TestData)
with pytest.raises(io_registry.IORegistryError):
io_registry.get_writer('test2', TestData)
def test_register_writer():
io_registry.register_writer('test1', TestData, empty_writer)
io_registry.register_writer('test2', TestData, empty_writer)
assert io_registry.get_writer('test1', TestData) == empty_writer
assert io_registry.get_writer('test2', TestData) == empty_writer
io_registry.unregister_writer('test1', TestData)
with pytest.raises(io_registry.IORegistryError):
io_registry.get_writer('test1', TestData)
assert io_registry.get_writer('test2', TestData) == empty_writer
io_registry.unregister_writer('test2', TestData)
with pytest.raises(io_registry.IORegistryError):
io_registry.get_writer('test2', TestData)
def test_unregister_writer_invalid():
with pytest.raises(io_registry.IORegistryError) as exc:
io_registry.unregister_writer('test', TestData)
assert str(exc.value
) == "No writer defined for format 'test' and class 'TestData'"
def test_unregister_writer_invalid():
with pytest.raises(io_registry.IORegistryError) as exc:
io_registry.unregister_writer('test', TestData)
assert str(exc.value) == "No writer defined for format 'test' and class 'TestData'"
def teardown_module(module):
"""clean up module after tests."""
io_registry.unregister_reader("json", Cosmology)
io_registry.unregister_writer("json", Cosmology)
io_registry.unregister_identifier("json", Cosmology)
def teardown_module(module):
"""clean up module after tests."""
with pytest.warns(FutureWarning): # idk
io_registry.unregister_reader("json", Cosmology)
io_registry.unregister_writer("json", Cosmology)
io_registry.unregister_identifier("json", Cosmology)
