[fbsync] Added missing assert (#6446)

Reviewed By: datumbox

Differential Revision: D39013668

fbshipit-source-id: d8661107e2af465243c73d86a265bd396327445c

Author: NicolasHug

test/test_prototype_transforms.py

@@ -793,7 +793,7 @@ def test__transform(self, p, transform_cls, func_op_name, kwargs, mocker):
         if p > 0.0:
             fn.assert_called_once_with(inpt, **kwargs)
         else:
-            fn.call_count == 0
+            assert fn.call_count == 0
 
 
 class TestRandomPerspective:
@@ -1014,7 +1014,7 @@ def test__transform(self, p, inpt_type, mocker):
         if p > 0.0:
             fn.assert_called_once_with(erase_image_tensor_inpt, **params)
         else:
-            fn.call_count == 0
+            assert fn.call_count == 0
 
 
 class TestTransform:
@@ -1050,7 +1050,7 @@ def test__transform(self, inpt_type, mocker):
         transform = transforms.ToImageTensor()
         transform(inpt)
         if inpt_type in (features.BoundingBox, str, int):
-            fn.call_count == 0
+            assert fn.call_count == 0
         else:
             fn.assert_called_once_with(inpt, copy=transform.copy)
 
@@ -1067,7 +1067,7 @@ def test__transform(self, inpt_type, mocker):
         transform = transforms.ToImagePIL()
         transform(inpt)
         if inpt_type in (features.BoundingBox, str, int):
-            fn.call_count == 0
+            assert fn.call_count == 0
         else:
             fn.assert_called_once_with(inpt, copy=transform.copy)
 
@@ -1085,7 +1085,7 @@ def test__transform(self, inpt_type, mocker):
             transform = transforms.ToPILImage()
         transform(inpt)
         if inpt_type in (PIL.Image.Image, features.BoundingBox, str, int):
-            fn.call_count == 0
+            assert fn.call_count == 0
         else:
             fn.assert_called_once_with(inpt, mode=transform.mode)
 
@@ -1103,7 +1103,7 @@ def test__transform(self, inpt_type, mocker):
             transform = transforms.ToTensor()
         transform(inpt)
         if inpt_type in (features.Image, torch.Tensor, features.BoundingBox, str, int):
-            fn.call_count == 0
+            assert fn.call_count == 0
         else:
             fn.assert_called_once_with(inpt)