def test_hdf5_protocol_1(): if 'proto1.h5' in os.listdir('.'): os.remove('proto1.h5') # Test material writing leu = Material({'U235': 0.04, 'U238': 0.96}, 4.2, 2.72, 1.0) leu.attrs['comment'] = 'first light' leu.write_hdf5('proto1.h5', chunksize=10) for i in range(2, 11): leu = Material({'U235': 0.04, 'U238': 0.96}, i * 4.2, 2.72, 1.0 * i) leu.attrs['comment'] = 'fire in the disco - {0}'.format(i) leu.write_hdf5('proto1.h5') # Loads with protocol 1 now. m = Material() m.from_hdf5('proto1.h5', '/material', -3, 1) assert_equal(m.density, 2.72) assert_equal(m.atoms_per_mol, 8.0) assert_equal(m.mass, 33.6) assert_equal(m.comp, {922350: 0.04, 922380: 0.96}) assert_equal(m.attrs['comment'], 'fire in the disco - 8') m = from_hdf5('proto1.h5', '/material', 3, 1) assert_equal(m.density, 2.72) assert_equal(m.atoms_per_mol, 4.0) assert_equal(m.mass, 16.8) assert_equal(m.comp, {922350: 0.04, 922380: 0.96}) assert_equal(m.attrs['comment'], 'fire in the disco - 4') os.remove('proto1.h5')
def test_hdf5_protocol_1(): if 'proto1.h5' in os.listdir('.'): os.remove('proto1.h5') # Test material writing leu = Material({'U235': 0.04, 'U238': 0.96}, 4.2, 2.72, 1.0) leu.metadata['comment'] = 'first light' leu.write_hdf5('proto1.h5', chunksize=10) for i in range(2, 11): leu = Material({'U235': 0.04, 'U238': 0.96}, i*4.2, 2.72, 1.0*i) leu.metadata['comment'] = 'fire in the disco - {0}'.format(i) leu.write_hdf5('proto1.h5') # Loads with protocol 1 now. m = Material() m.from_hdf5('proto1.h5', '/material', -3, 1) assert_equal(m.density, 2.72) assert_equal(m.atoms_per_molecule, 8.0) assert_equal(m.mass, 33.6) assert_equal(m.comp, {922350000: 0.04, 922380000: 0.96}) assert_equal(m.metadata['comment'], 'fire in the disco - 8') m = from_hdf5('proto1.h5', '/material', 3, 1) assert_equal(m.density, 2.72) assert_equal(m.atoms_per_molecule, 4.0) assert_equal(m.mass, 16.8) assert_equal(m.comp, {922350000: 0.04, 922380000: 0.96}) assert_equal(m.metadata['comment'], 'fire in the disco - 4') os.remove('proto1.h5')
def test_hdf5_protocol_1(): if 'proto1.h5' in os.listdir('.'): os.remove('proto1.h5') # Test material writing leu = Material({'U235': 0.04, 'U238': 0.96}, 4.2, "LEU", 1.0) leu.write_hdf5('proto1.h5', chunksize=10) for i in range(2, 11): leu = Material({'U235': 0.04, 'U238': 0.96}, i*4.2, "LEU", 1.0*i) leu.write_hdf5('proto1.h5') # Loads with protocol 1 now. m = Material() m.from_hdf5('proto1.h5', '/material', -3, 1) assert_equal(m.name, 'LEU') assert_equal(m.atoms_per_mol, 8.0) assert_equal(m.mass, 33.6) assert_equal(m.comp, {922350: 0.04, 922380: 0.96}) m = from_hdf5('proto1.h5', '/material', 3, 1) assert_equal(m.name, 'LEU') assert_equal(m.atoms_per_mol, 4.0) assert_equal(m.mass, 16.8) assert_equal(m.comp, {922350: 0.04, 922380: 0.96}) os.remove('proto1.h5')
def test_from_hdf5_func_protocol_0(): mat = from_hdf5("mat.h5", "/mat", protocol=0) assert_equal(mat.mass, 0.0) assert_equal(mat.comp, {922350: 0.0, 942390: 0.0}) mat = from_hdf5("mat.h5", "/mat", 0, 0) assert_equal(mat.mass, 1.0) assert_equal(mat.comp, {922350: 1.0, 942390: 0.0}) mat = from_hdf5("mat.h5", "/mat", 1, 0) assert_equal(mat.mass, 0.5) assert_equal(mat.comp, {922350: 0.75, 942390: 0.25}) mat = from_hdf5("mat.h5", "/mat", 2, 0) assert_equal(mat.mass, 0.0) assert_equal(mat.comp, {922350: 0.0, 942390: 0.0}) mat = from_hdf5("mat.h5", "/mat", -1, 0) assert_equal(mat.mass, 0.0) assert_equal(mat.comp, {922350: 0.0, 942390: 0.0}) mat = from_hdf5("mat.h5", "/mat", -2, 0) assert_equal(mat.mass, 0.5) assert_equal(mat.comp, {922350: 0.75, 942390: 0.25}) mat = from_hdf5("mat.h5", "/mat", -3, 0) assert_equal(mat.mass, 1.0) assert_equal(mat.comp, {922350: 1.0, 942390: 0.0})
def test_from_hdf5_func_protocol_0(): mat = from_hdf5("mat.h5", "/mat", protocol=0) assert_equal(mat.mass, 0.0) assert_equal(mat.comp, {922350000: 0.0, 942390000: 0.0}) mat = from_hdf5("mat.h5", "/mat", 0, 0) assert_equal(mat.mass, 1.0) assert_equal(mat.comp, {922350000: 1.0, 942390000: 0.0}) mat = from_hdf5("mat.h5", "/mat", 1, 0) assert_equal(mat.mass, 0.5) assert_equal(mat.comp, {922350000: 0.75, 942390000: 0.25}) mat = from_hdf5("mat.h5", "/mat", 2, 0) assert_equal(mat.mass, 0.0) assert_equal(mat.comp, {922350000: 0.0, 942390000: 0.0}) mat = from_hdf5("mat.h5", "/mat", -1, 0) assert_equal(mat.mass, 0.0) assert_equal(mat.comp, {922350000: 0.0, 942390000: 0.0}) mat = from_hdf5("mat.h5", "/mat", -2, 0) assert_equal(mat.mass, 0.5) assert_equal(mat.comp, {922350000: 0.75, 942390000: 0.25}) mat = from_hdf5("mat.h5", "/mat", -3, 0) assert_equal(mat.mass, 1.0) assert_equal(mat.comp, {922350000: 1.0, 942390000: 0.0})