test deprecated call

Author: DuYicong515

tests/accelerators/test_tpu.py

@@ -102,11 +102,6 @@ def test_accelerator_tpu(accelerator, devices):
     assert isinstance(trainer.accelerator, TPUAccelerator)
     assert isinstance(trainer.strategy, TPUSpawnStrategy)
     assert trainer.num_devices == 8
-    with pytest.deprecated_call(
-        match="`Trainer.tpu_cores` is deprecated in v1.6 and will be removed in v1.8. "
-        "Please use `Trainer.devices` instead."
-    ):
-        trainer.tpu_cores == 8
 
 
 @RunIf(tpu=True)
@@ -124,6 +119,7 @@ def test_accelerator_tpu_with_tpu_cores_priority():
 @RunIf(tpu=True)
 def test_set_devices_if_none_tpu():
     trainer = Trainer(accelerator="tpu", tpu_cores=8)
+    assert isinstance(trainer.accelerator, TPUAccelerator)
     assert trainer.num_devices == 8
 
 

tests/deprecated_api/test_remove_1-8.py

@@ -1106,3 +1106,14 @@ def on_save_checkpoint(self, trainer, pl_module, checkpoint):
 
     trainer.callbacks = [TestCallbackSaveHookOverride()]
     trainer.save_checkpoint(tmpdir + "/pathok.ckpt")
+
+
+def test_trainer_tpu_cores(monkeypatch):
+    monkeypatch.setattr(pytorch_lightning.accelerators.tpu.TPUAccelerator, "is_available", lambda: True)
+    monkeypatch.setattr(pytorch_lightning.accelerators.tpu.TPUAccelerator, "parse_devices", lambda: 8)
+    trainer = Trainer(accelerator="TPU", devices=8)
+    with pytest.deprecated_call(
+        match="`Trainer.tpu_cores` is deprecated in v1.6 and will be removed in v1.8. "
+        "Please use `Trainer.devices` instead."
+    ):
+        trainer.tpu_cores == 8