AtHeartEngineer/kaipy
1
0
Fork 0
mirror of https://github.com/JHUAPL/kaipy.git synced 2026-01-09 22:37:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for msphViz

Browse Source
This commit is contained in:
wiltbemj
2025-02-19 07:26:29 -07:00
parent ee35fc8b52
commit 9fc97a33fe
2 changed files with 174 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tests/conftest.py
View File

@@ -18,11 +18,13 @@ def gamera_pipe(tmpdir):
fdir = tmpdir.mkdir("data")
ftag = "test"
file_path = fdir.join("test.gam.h5")
print(f'file_path: {file_path}')
mjds = []
with h5py.File(file_path, 'w') as f:
f.create_dataset("X", data=np.zeros((Ni+1, Nj+1, Nk+1)))
f.create_dataset("Y", data=np.zeros((Ni+1, Nj+1, Nk+1)))
f.create_dataset("Z", data=np.zeros((Ni+1, Nj+1, Nk+1)))
f.create_dataset("X", data=np.linspace(-100, 100, Ni+1).reshape(Ni+1, 1, 1).repeat(Nj+1, axis=1).repeat(Nk+1, axis=2).T)
f.create_dataset("Y", data=np.linspace(-100, 100, Nj+1).reshape(1, Nj+1, 1).repeat(Ni+1, axis=0).repeat(Nk+1, axis=2).T)
f.create_dataset("Z", data=np.linspace(-100, 100, Nk+1).reshape(1, 1, Nk+1).repeat(Ni+1, axis=0).repeat(Nj+1, axis=1).T)
f.create_dataset("dV", data=np.zeros((Ni, Nj, Nk)).T)
for i in range(3):
grp = f.create_group("Step#{}".format(i))
grp.attrs['time'] = np.double(i)
@@ -30,14 +32,17 @@ def gamera_pipe(tmpdir):
grp.attrs['MJD'] = mjd
mjds.append(mjd)
grp.attrs['timestep'] = i
grp.create_dataset("D", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("P", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("Bx", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("By", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("Bz", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("Vx", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("Vy", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("Vz", data=np.zeros((Ni, Nj, Nk)))
grp.create_dataset("D", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("P", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Bx", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("By", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Bz", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Vx", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Vy", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Vz", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Jx", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Jy", data=np.zeros((Ni, Nj, Nk)).T)
grp.create_dataset("Jz", data=np.zeros((Ni, Nj, Nk)).T)
file_path = fdir.join("test.mix.h5")
lat = np.arange(Nlat+1)
@@ -57,5 +62,13 @@ def gamera_pipe(tmpdir):
grp.create_dataset("Potential SOUTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Field-aligned current NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Field-aligned current SOUTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Pedersen conductance NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Pedersen conductance SOUTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Hall conductance NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Hall conductance SOUTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Average energy NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Average energy SOUTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Number flux NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Number flux SOUTH", data=np.zeros((Nlat, Nlon)))
return GamsphPipe(str(fdir), ftag, doFast=True, uID="Earth")