remove deprecated assertDictContainsSubset

PiperOrigin-RevId: 716228305 Change-Id: I95f3d853787621f966ba421af91ca9df5db9c9f5 GitOrigin-RevId: 5834b45a25cf8672fabc292063e81a4ca96c0103
google-deepmind · Jan 23, 2025 · 7deae30 · 7deae30
1 parent 3d14a33
commit 7deae30
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/xmanager/contrib/tensorboard_test.py b/xmanager/contrib/tensorboard_test.py
@@ -124,16 +124,14 @@ def test_get_tensorboard_job_args(
     provider = tensorboard.TensorboardProvider
 
     args = provider.get_tensorboard_job_args(log_dir, port, additional_args)
-    self.assertDictContainsSubset(
-        dict(
-            {
-                'logdir': log_dir,
-                'port': port or provider.DEFAULT_TENSORBOARD_PORT,
-            },
-            **(additional_args if additional_args else {}),
-        ),
-        args,
+    expected = dict(
+        {
+            'logdir': log_dir,
+            'port': port or provider.DEFAULT_TENSORBOARD_PORT,
+        },
+        **(additional_args if additional_args else {}),
     )
+    self.assertEqual(args, {**args, **expected})
 
 
 if __name__ == '__main__':