[fbsync] Use torch.testing.assert_close in test_transforms_video.py (#…

…3886)

Summary: Co-authored-by: Philip Meier <github.pmeier@posteo.de>

Reviewed By: vincentqb, cpuhrsch

Differential Revision: D28679968

fbshipit-source-id: cc92cfa4eb77ad740f671a49c4a4aa0d92764230

test/test_transforms_video.py

@@ -4,6 +4,7 @@
 import random
 import numpy as np
 import warnings
+from _assert_utils import assert_equal
 
 try:
     from scipy import stats
@@ -120,7 +121,7 @@ def samples_from_standard_normal(tensor):
         # Checking the optional in-place behaviour
         tensor = torch.rand((3, 128, 16, 16))
         tensor_inplace = transforms.NormalizeVideo((0.5, 0.5, 0.5), (0.5, 0.5, 0.5), inplace=True)(tensor)
-        self.assertTrue(torch.equal(tensor, tensor_inplace))
+        assert_equal(tensor, tensor_inplace)
 
         transforms.NormalizeVideo((0.5, 0.5, 0.5), (0.5, 0.5, 0.5), inplace=True).__repr__()