Skip to content
This repository has been archived by the owner on Jan 10, 2025. It is now read-only.

Commit

Permalink
test on statistics
Browse files Browse the repository at this point in the history
floriankrb committed Mar 11, 2024
1 parent ae0ec9e commit 6f6777c
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions tests/create/test_create.py
Original file line number Diff line number Diff line change
@@ -90,6 +90,19 @@ def compare_datasets(a, b):
assert max_delta == 0.0, (date, param, a_, b_, a_ - b_, max_delta)


def compare_statistics(ds1, ds2):
vars1 = ds1.variables
vars2 = ds2.variables
assert len(vars1) == len(vars2)
for v1, v2 in zip(vars1, vars2):
idx1 = ds1.name_to_index[v1]
idx2 = ds2.name_to_index[v2]
assert (ds1.statistics["mean"][idx1] == ds2.statistics["mean"][idx2]).all()
assert (ds1.statistics["stdev"][idx1] == ds2.statistics["stdev"][idx2]).all()
assert (ds1.statistics["maximum"][idx1] == ds2.statistics["maximum"][idx2]).all()
assert (ds1.statistics["minimum"][idx1] == ds2.statistics["minimum"][idx2]).all()


class Comparer:
def __init__(self, name, output_path=None, reference_path=None):
self.name = name
@@ -106,8 +119,7 @@ def __init__(self, name, output_path=None, reference_path=None):
def compare(self):
compare_dot_zattrs(self.z_output.attrs, self.z_reference.attrs)
compare_datasets(self.ds_output, self.ds_reference)
# not implemented :
# compare_statistics(self.z_output, self.z_reference)
compare_statistics(self.ds_output, self.ds_reference)


@pytest.mark.parametrize("name", NAMES)

0 comments on commit 6f6777c

Please sign in to comment.