fbcode//torchx/schedulers/test (#940)

Summary: Pull Request resolved: #940 Differential Revision: D60508215
pytorch · Sep 10, 2024 · 5c730f7 · 5c730f7
1 parent 66733b7
commit 5c730f7
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/torchx/schedulers/test/aws_batch_scheduler_test.py b/torchx/schedulers/test/aws_batch_scheduler_test.py
@@ -756,7 +756,7 @@ def worker() -> None:
 
     def test_resource_requirement_from_resource(self) -> None:
         cpu_resource = Resource(cpu=2, memMB=1024, gpu=0)
-        self.assertEquals(
+        self.assertEqual(
             [
                 {"type": "VCPU", "value": "2"},
                 {"type": "MEMORY", "value": "1024"},
@@ -765,7 +765,7 @@ def test_resource_requirement_from_resource(self) -> None:
         )
 
         zero_cpu_resource = Resource(cpu=0, memMB=1024, gpu=0)
-        self.assertEquals(
+        self.assertEqual(
             [
                 {"type": "VCPU", "value": "1"},
                 {"type": "MEMORY", "value": "1024"},
@@ -774,7 +774,7 @@ def test_resource_requirement_from_resource(self) -> None:
         )
 
         gpu_resource = Resource(cpu=1, memMB=1024, gpu=2)
-        self.assertEquals(
+        self.assertEqual(
             [
                 {"type": "VCPU", "value": "1"},
                 {"type": "MEMORY", "value": "1024"},

diff --git a/torchx/schedulers/test/kubernetes_mcad_scheduler_test.py b/torchx/schedulers/test/kubernetes_mcad_scheduler_test.py
@@ -167,8 +167,8 @@ def test_create_scheduler(self) -> None:
         self.assertIsInstance(
             scheduler, kubernetes_mcad_scheduler.KubernetesMCADScheduler
         )
-        self.assertEquals(client, scheduler._client)
-        self.assertEquals(docker_client, scheduler._docker_client)
+        self.assertEqual(client, scheduler._client)
+        self.assertEqual(docker_client, scheduler._docker_client)
 
     def test_app_to_resource_resolved_macros(self) -> None:
         app = _test_app()

diff --git a/torchx/schedulers/test/kubernetes_scheduler_test.py b/torchx/schedulers/test/kubernetes_scheduler_test.py
@@ -99,8 +99,8 @@ def test_create_scheduler(self) -> None:
         docker_client = MagicMock
         scheduler = create_scheduler("foo", client=client, docker_client=docker_client)
         self.assertIsInstance(scheduler, kubernetes_scheduler.KubernetesScheduler)
-        self.assertEquals(scheduler._docker_client, docker_client)
-        self.assertEquals(scheduler._client, client)
+        self.assertEqual(scheduler._docker_client, docker_client)
+        self.assertEqual(scheduler._client, client)
 
     def test_app_to_resource_resolved_macros(self) -> None:
         app = _test_app()