def test_h5_io(): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) ds = fake_random_ds(64, nprocs=1, length_unit=10) warr = ds.arr(np.random.random((256, 256)), 'code_length') warr.write_hdf5('test.h5') iarr = YTArray.from_hdf5('test.h5') assert_equal(warr, iarr) assert_equal(warr.units.registry['code_length'], iarr.units.registry['code_length']) warr.write_hdf5('test.h5', dataset_name="test_dset", group_name='/arrays/test_group') giarr = YTArray.from_hdf5('test.h5', dataset_name="test_dset", group_name='/arrays/test_group') assert_equal(warr, giarr) os.chdir(curdir) shutil.rmtree(tmpdir)
def test_h5_io(): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) ds = fake_random_ds(64, nprocs=1, length_unit=10) warr = ds.arr(np.random.random((256, 256)), 'code_length') warr.write_hdf5('test.h5') iarr = YTArray.from_hdf5('test.h5') yield assert_equal, warr, iarr yield assert_equal, warr.units.registry['code_length'], iarr.units.registry['code_length'] os.chdir(curdir) shutil.rmtree(tmpdir)
def test_h5_io(): tmpdir = tempfile.mkdtemp() curdir = os.getcwd() os.chdir(tmpdir) ds = fake_random_ds(64, nprocs=1, length_unit=10) warr = ds.arr(np.random.random((256, 256)), 'code_length') warr.write_hdf5('test.h5') iarr = YTArray.from_hdf5('test.h5') yield assert_equal, warr, iarr yield assert_equal, warr.units.registry[ 'code_length'], iarr.units.registry['code_length'] os.chdir(curdir) shutil.rmtree(tmpdir)