From 7ab2cad9383fab181f21d643546804d4e7fdbb85 Mon Sep 17 00:00:00 2001 From: cookieenick Date: Mon, 8 Dec 2025 08:17:29 -0700 Subject: [PATCH] Updated test mix generator --- tests/conftest.py | 6 ++++++ tests/gamera/test_mspViz.py | 10 +++++----- 2 files changed, 11 insertions(+), 5 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index f52f291..b282eae 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -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 diff --git a/tests/gamera/test_mspViz.py b/tests/gamera/test_mspViz.py index fdec7e4..c2ffc20 100644 --- a/tests/gamera/test_mspViz.py +++ b/tests/gamera/test_mspViz.py @@ -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):