def test_combine_along_y(self, tmpdir_factory, bout_xyt_example_files): path = bout_xyt_example_files(tmpdir_factory, nxpe=1, nype=3, nt=1, syn_data_type='stepped') actual, metadata = _auto_open_mfboutdataset(datapath=path) bout_ds = create_bout_ds expected = concat([bout_ds(0), bout_ds(1), bout_ds(2)], dim='y') xrt.assert_equal(actual.load(), expected.drop(METADATA_VARS))
def test_combine_along_xy(self, tmpdir_factory, bout_xyt_example_files): path = bout_xyt_example_files(tmpdir_factory, nxpe=4, nype=3, nt=1, syn_data_type='stepped') actual, metadata = _auto_open_mfboutdataset(datapath=path) bout_ds = create_bout_ds line1 = concat([bout_ds(0), bout_ds(1), bout_ds(2), bout_ds(3)], dim='x') line2 = concat([bout_ds(4), bout_ds(5), bout_ds(6), bout_ds(7)], dim='x') line3 = concat([bout_ds(8), bout_ds(9), bout_ds(10), bout_ds(11)], dim='x') expected = concat([line1, line2, line3], dim='y') xrt.assert_equal(actual.load(), expected.drop(METADATA_VARS))
def test_multiple_files_along_x(self): var = 'n' expected = collect(var, path='./tests/data/dump_files/', prefix='BOUT.dmp', xguards=False) ds, metadata = _auto_open_mfboutdataset( './tests/data/dump_files/BOUT.dmp.*.nc') actual = ds[var].values assert expected.shape == actual.shape npt.assert_equal(actual, expected)
def test_single_file(self): var = 'n' expected = collect(var, path='./tests/data/dump_files/single', prefix='equilibrium', xguards=False) ds, metadata = _auto_open_mfboutdataset( './tests/data/dump_files/single/equilibrium.nc') print(ds) actual = ds[var].values assert expected.shape == actual.shape npt.assert_equal(actual, expected)
def test_single_file(self, tmpdir_factory, bout_xyt_example_files): path = bout_xyt_example_files(tmpdir_factory, nxpe=1, nype=1, nt=1) actual, metadata = _auto_open_mfboutdataset(datapath=path) expected = create_bout_ds() xrt.assert_equal(actual.load(), expected.drop(METADATA_VARS))