fix tests --- now need divisibility

mehta-lab · Feb 10, 2025 · bb39e9d · bb39e9d
1 parent 16c87a7
commit bb39e9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_filter.py b/tests/test_filter.py
@@ -33,11 +33,11 @@ def test_stretched_multiply():
 
     # Test that output dims are correct
     rand_array_3x3x3 = torch.rand((3, 3, 3))
-    rand_array_100x100x100 = torch.rand((100, 100, 100))
+    rand_array_100x100x100 = torch.rand((99, 99, 99))
     result = filter.stretched_multiply(
         rand_array_3x3x3, rand_array_100x100x100
     )
-    assert result.shape == (100, 100, 100)
+    assert result.shape == (99, 99, 99)
 
 
 def test_stretched_multiply_incompatible_dims():