JHUAPL/kaipy
1
0
Fork 0
mirror of https://github.com/JHUAPL/kaipy.git synced 2026-01-06 22:44:03 -05:00
Code Issues Packages Projects Releases 2 Wiki Activity

Updated test mix generator

Browse Source
This commit is contained in:
cookieenick
2025-12-08 08:17:29 -07:00
parent f208b7e1c6
commit 7ab2cad938
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/conftest.py
View File

@@ -47,6 +47,12 @@ def write_mix_h5(file_path,mjds=None):
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)))
grp.create_dataset("RCM beta NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("RCM beta SOUTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Mono potential drop NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Mono potential drop SOUTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Auroral model type NORTH", data=np.zeros((Nlat, Nlon)))
grp.create_dataset("Auroral model type SOUTH", data=np.zeros((Nlat, Nlon)))
return file_path

10
tests/gamera/test_mspViz.py
View File

@@ -7,7 +7,7 @@ import argparse
from argparse import RawTextHelpFormatter
import kaipy.remix.remix as remix
from kaipy.gamera.msphViz import AddSizeArgs, GetSizeBds, PlotEqErrAbs, PlotEqErrRel, PlotLogicalErrAbs, PlotLogicalErrRel, CalcTotalErrAbs, CalcTotalErrRel, PlotEqB, PlotMerid, PlotJyXZ, PlotEqEphi, PlotMPI, AddIonBoxes, plotPlane, plotXY, plotXZ
from kaipy.gamera.msphViz import AddSizeArgs, GetSizeBds, PlotErrAbs, PlotErrRel, PlotLogicalErrAbs, PlotLogicalErrRel, CalcTotalErrAbs, CalcTotalErrRel, PlotEqB, PlotMerid, PlotJyXZ, PlotEqEphi, PlotMPI, AddIonBoxes, plotPlane, plotXY, plotXZ
@pytest.fixture
@@ -26,22 +26,22 @@ def test_GetSizeBds():
result = GetSizeBds(args)
assert result == [-100.0, 20.0, -60.0, 60.0]
def test_PlotEqErrAbs(gamera_pipe):
def test_PlotErrAbs(gamera_pipe):
fig, ax = plt.subplots()
gsphP = gamera_pipe
gsphO = gamera_pipe
fieldNames = ['Bz']
xyBds = [-100, 100, -100, 100]
result = PlotEqErrAbs(gsphP, gsphO, 0, xyBds, ax, fieldNames)
result = PlotErrAbs(gsphP, gsphO, 0, xyBds, ax, fieldNames)
assert isinstance(result, np.ndarray)
def test_PlotEqErrRel(gamera_pipe):
def test_PlotErrRel(gamera_pipe):
fig, ax = plt.subplots()
gsphP = gamera_pipe
gsphO = gamera_pipe
fieldNames = ['Bz']
xyBds = [-100, 100, -100, 100]
result = PlotEqErrRel(gsphP, gsphO, 0, xyBds, ax, fieldNames)
result = PlotErrRel(gsphP, gsphO, 0, xyBds, ax, fieldNames)
assert isinstance(result, np.ndarray)
def test_PlotLogicalErrAbs(gamera_pipe):