tests/test_core.py: fix failing test

samuela · Jan 5, 2025 · 13829e5 · 13829e5
1 parent 6faa450
commit 13829e5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -15,9 +15,10 @@ def test_arange():
 
 
 def test_empty():
-  t2j_function_test(lambda: torch.empty(()), [])
-  t2j_function_test(lambda: torch.empty(2), [])
-  t2j_function_test(lambda: torch.empty((2, 3)), [])
+  # torch.empty returns uninitialized values, so we need to multiply by 0 for deterministic, testable behavior.
+  t2j_function_test(lambda: 0 * torch.empty(()), [])
+  t2j_function_test(lambda: 0 * torch.empty(2), [])
+  t2j_function_test(lambda: 0 * torch.empty((2, 3)), [])
 
 
 def test_ones():