Add new assert_equal test util based on torch.testing.assert_close (#…

…3872) Co-authored-by: Philip Meier <github.pmeier@posteo.de>
pytorch · May 21, 2021 · 89cb908 · 89cb908
1 parent 231f80d
commit 89cb908
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/test/_assert_utils.py b/test/_assert_utils.py
@@ -0,0 +1,11 @@
+"""This is a temporary module and should be removed as soon as torch.testing.assert_equal is supported."""
+# TODO: remove this as soon torch.testing.assert_equal is supported
+
+import functools
+
+import torch.testing
+
+__all__ = ["assert_equal"]
+
+
+assert_equal = functools.partial(torch.testing.assert_close, rtol=0, atol=0)