fbcode//torchx/schedulers/test [A]

itamaro · web-flow · commit 0467f3355e9f · 2025-01-31T11:52:18.000-08:00
Differential Revision: D68445042 Pull Request resolved: #1004
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"},

Original file line number	Diff line number	Diff line change
`@@ -756,7 +756,7 @@ def worker() -> None:`
`756`	`756`
`757`	`757`	`def test_resource_requirement_from_resource(self) -> None:`
`758`	`758`	`cpu_resource = Resource(cpu=2, memMB=1024, gpu=0)`
`759`		`- self.assertEquals(`
	`759`	`+ self.assertEqual(`
`760`	`760`	`[`
`761`	`761`	`{"type": "VCPU", "value": "2"},`
`762`	`762`	`{"type": "MEMORY", "value": "1024"},`
`@@ -765,7 +765,7 @@ def test_resource_requirement_from_resource(self) -> None:`
`765`	`765`	`)`
`766`	`766`
`767`	`767`	`zero_cpu_resource = Resource(cpu=0, memMB=1024, gpu=0)`
`768`		`- self.assertEquals(`
	`768`	`+ self.assertEqual(`
`769`	`769`	`[`
`770`	`770`	`{"type": "VCPU", "value": "1"},`
`771`	`771`	`{"type": "MEMORY", "value": "1024"},`
`@@ -774,7 +774,7 @@ def test_resource_requirement_from_resource(self) -> None:`
`774`	`774`	`)`
`775`	`775`
`776`	`776`	`gpu_resource = Resource(cpu=1, memMB=1024, gpu=2)`
`777`		`- self.assertEquals(`
	`777`	`+ self.assertEqual(`
`778`	`778`	`[`
`779`	`779`	`{"type": "VCPU", "value": "1"},`
`780`	`780`	`{"type": "MEMORY", "value": "1024"},`