fbcode//mobile-vision/d2go/tests

Summary: Pull Request resolved: https://github.com/facebookresearch/d2go/pull/648

Differential Revision: D52792992

fbshipit-source-id: d3a64f3b306ea024ec072eaa6327446a84e2d83c

tests/export/test_torchscript.py

@@ -34,24 +34,24 @@ class TestTorchscriptExportMethods(unittest.TestCase):
         export_method_string = "torchscript_mobile"
         new_export_kwargs = try_mock_export(export_method_string)
         self.assertIn("mobile_optimization", new_export_kwargs)
-        self.assertEquals(
+        self.assertEqual(
             type(new_export_kwargs["mobile_optimization"]),
             MobileOptimizationConfig,
         )
-        self.assertEquals(new_export_kwargs["mobile_optimization"].backend, "CPU")
+        self.assertEqual(new_export_kwargs["mobile_optimization"].backend, "CPU")
 
         export_method_string = "torchscript_mobile-metal"
         new_export_kwargs = try_mock_export(export_method_string)
-        self.assertEquals(new_export_kwargs["mobile_optimization"].backend, "metal")
+        self.assertEqual(new_export_kwargs["mobile_optimization"].backend, "metal")
 
         export_method_string = "torchscript_mobile-vulkan"
         new_export_kwargs = try_mock_export(export_method_string)
-        self.assertEquals(new_export_kwargs["mobile_optimization"].backend, "vulkan")
+        self.assertEqual(new_export_kwargs["mobile_optimization"].backend, "vulkan")
 
         export_method_string = "torchscript_mobile@tracing"
         new_export_kwargs = try_mock_export(export_method_string)
-        self.assertEquals(new_export_kwargs["jit_mode"], "trace")
+        self.assertEqual(new_export_kwargs["jit_mode"], "trace")
 
         export_method_string = "torchscript_mobile@scripting"
         new_export_kwargs = try_mock_export(export_method_string)
-        self.assertEquals(new_export_kwargs["jit_mode"], "script")
+        self.assertEqual(new_export_kwargs["jit_mode"], "script")