Skip to content
Snippets Groups Projects
Commit aa5a3c17 authored by fraserwg's avatar fraserwg
Browse files

updated the tests

parent d92a415e
No related branches found
No related tags found
1 merge request!11Updates to calculate_xr and addition of tests
......@@ -63,16 +63,24 @@ def test_xr_crop_tgrid(tgrid, request):
& (tgrid["clat"] > -5) & (tgrid["clat"] < 5),
drop=True).astype("int32")
# This checks ireg_c is as expected
assert ireg_c.sum() == 301614
assert ireg_c.prod() == -8253145384319188992
cropped_tgrid = pyic.xr_crop_tgrid(tgrid, ireg_c)
# Check ireg_[cev] is present
for point in "ev":
for point in "cev":
assert f"ireg_{point}" in cropped_tgrid.keys()
# Check ncells == len(ireg_c)
assert cropped_tgrid.dims["cell"] == ireg_c.sizes["cell"]
# Check ireg_[ev] is correct
# Check ireg_[cev] is correct
# Ideally we would hash the array and compare, but this will probably do
assert cropped_tgrid["ireg_c"].sum() == 301614
assert cropped_tgrid["ireg_c"].prod() == -8253145384319188992
assert cropped_tgrid["ireg_e"].sum() == 839941
assert cropped_tgrid["ireg_e"].prod() == 0
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment