MCPcopy Create free account
hub / github.com/PowerGridModel/power-grid-model / compare_row_data

Function compare_row_data

tests/unit/test_internal_utils.py:876–896  ·  view source on GitHub ↗
(actual_row_data, desired_row_data)

Source from the content-addressed store, hash-verified

874
875
876def compare_row_data(actual_row_data, desired_row_data):
877 assert actual_row_data.keys() == desired_row_data.keys()
878
879 for comp_name in actual_row_data:
880 actual_component = actual_row_data[comp_name]
881 desired_component = desired_row_data[comp_name]
882 if is_sparse(actual_component):
883 assert actual_component.keys() == desired_component.keys()
884 assert np.array_equal(actual_component["indptr"], desired_component["indptr"])
885 actual_component = actual_component["data"]
886 desired_component = desired_component["data"]
887 assert actual_component.dtype == desired_component.dtype
888 assert actual_component.shape == desired_component.shape
889
890 for field in actual_component.dtype.names:
891 actual_attr = actual_component[field]
892 desired_attr = desired_component[field]
893 nan_a = np.isnan(actual_attr)
894 nan_b = np.isnan(desired_attr)
895 np.testing.assert_array_equal(nan_a, nan_b)
896 np.testing.assert_allclose(actual_attr[~nan_a], desired_attr[~nan_b], rtol=1e-15)
897
898
899def test_dense_row_data_to_from_col_data(row_data):

Callers 1

Calls 1

is_sparseFunction · 0.90

Tested by

no test coverage detected