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)