Add key as error message in assert_dataset for floating value cases.

PiperOrigin-RevId: 557644482
google · Aug 16, 2023 · 6594a3c · 6594a3c
1 parent 5122776
commit 6594a3c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/seqio/test_utils.py b/seqio/test_utils.py
@@ -511,7 +511,9 @@ def _compare_dict(actual_dict, expected_dict):
           or isinstance(actual_value, np.ndarray)
           and np.issubdtype(actual_value.dtype, np.floating)
       ):
-        np.testing.assert_allclose(actual_value, expected_dict[key])
+        np.testing.assert_allclose(
+            actual_value, expected_dict[key], err_msg=key
+        )
       else:
         np.testing.assert_array_equal(
             actual_value, _maybe_as_bytes(expected_dict[key]), key