def test_write_object(): """Test writing of a object. """ tmpfile = tempfile.NamedTemporaryFile() storage = NetCDFStorage(tmpfile.name, mode='w') #use names we might encounter in simulation envname = 'vacuum' modname = 'ExpandedEnsembleSampler' varname = 'energy' view = NetCDFStorageView(storage, envname, modname) obj = { 0 : 0 } view.write_object('singleton', obj) for iteration in range(10): obj = { 'iteration' : iteration } view.write_object(varname, obj, iteration=iteration) for iteration in range(10): obj = storage.get_object(envname, modname, varname, iteration=iteration) assert ('iteration' in obj) assert (obj['iteration'] == iteration)
def test_write_object(): """Test writing of a object. """ tmpfile = tempfile.NamedTemporaryFile() storage = NetCDFStorage(tmpfile.name, mode='w') #use names we might encounter in simulation envname = 'vacuum' modname = 'ExpandedEnsembleSampler' varname = 'energy' view = NetCDFStorageView(storage, envname, modname) obj = {0: 0} view.write_object('singleton', obj) for iteration in range(10): obj = {'iteration': iteration} view.write_object(varname, obj, iteration=iteration) for iteration in range(10): obj = storage.get_object(envname, modname, varname, iteration=iteration) assert ('iteration' in obj) assert (obj['iteration'] == iteration)
def test_write_object(): """Test writing of a object. """ tmpfile = tempfile.NamedTemporaryFile() storage = NetCDFStorage(tmpfile.name, mode='w') view = NetCDFStorageView(storage, 'envname', 'modname') obj = { 0 : 0 } view.write_object('singleton', obj) for iteration in range(10): obj = { 'iteration' : iteration } view.write_object('varname', obj, iteration=iteration) for iteration in range(10): obj = storage.get_object('/envname/modname/varname', iteration=iteration) assert ('iteration' in obj) assert (obj['iteration'] == iteration)