Correct misscalculation

Author: anamatoso

tests/test_subtract_itemsd.py

@@ -34,7 +34,7 @@ def test_tensor_values(self):
         self.assertIn("sub_img", result)
         result["sub_img"] += 1
         assert_allclose(result["img1"], torch.tensor([[0, 1], [1, 2]], device=device))
-        assert_allclose(result["sub_img"], torch.tensor([[0, 0], [0, 0]], device=device))
+        assert_allclose(result["sub_img"], torch.tensor([[1, 1], [1, 1]], device=device))
 
     def test_metatensor_values(self):
         device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu:0")
@@ -48,15 +48,15 @@ def test_metatensor_values(self):
         self.assertEqual(result["img1"].meta, result["sub_img"].meta)
         result["sub_img"] += 1
         assert_allclose(result["img1"], torch.tensor([[0, 1], [1, 2]], device=device))
-        assert_allclose(result["sub_img"], torch.tensor([[0, 0], [0, 0]], device=device))
+        assert_allclose(result["sub_img"], torch.tensor([[1, 1], [1, 1]], device=device))
 
     def test_numpy_values(self):
         input_data = {"img1": np.array([[0, 1], [1, 2]]), "img2": np.array([[0, 1], [1, 2]])}
         result = SubtractItemsd(keys=["img1", "img2"], name="sub_img")(input_data)
         self.assertIn("sub_img", result)
         result["sub_img"] += 1
         np.testing.assert_allclose(result["img1"], np.array([[0, 1], [1, 2]]))
-        np.testing.assert_allclose(result["sub_img"], np.array([[0, 0], [0, 0]]))
+        np.testing.assert_allclose(result["sub_img"], np.array([[1, 1], [1, 1]]))
 
 if __name__ == "__main__":
     unittest.main()