def register(): """Register the h5s3 driver with h5py. See Also -------- :func:`h5s3.unregiser` """ h5py.register_driver('h5s3', set_fapl)
def test_unregister_driver(self): h5py.register_driver('new-driver', lambda plist: None) self.assertIn('new-driver', h5py.registered_drivers()) h5py.unregister_driver('new-driver') self.assertNotIn('new-driver', h5py.registered_drivers()) with self.assertRaises(ValueError) as e: fname = self.mktemp() h5py.File(fname, driver='new-driver') self.assertEqual(str(e.exception), 'Unknown driver type "new-driver"')
def test_register_driver(self): called_with = [None] def set_fapl(plist, *args, **kwargs): called_with[0] = args, kwargs return _drivers['sec2'](plist) h5py.register_driver('new-driver', set_fapl) self.assertIn('new-driver', h5py.registered_drivers()) fname = self.mktemp() h5py.File(fname, driver='new-driver', driver_arg_0=0, driver_arg_1=1) self.assertEqual( called_with, [((), {'driver_arg_0': 0, 'driver_arg_1': 1})], )