Exemplo n.º 1
0
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)
Exemplo n.º 2
0
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)
Exemplo n.º 3
0
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)