test for #1729

Lightning-AI · May 17, 2020 · 5245a82 · 5245a82
1 parent 981f50c
commit 5245a82
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_deprecated.py b/tests/test_deprecated.py
@@ -97,7 +97,8 @@ def test_tbd_remove_in_v0_9_0_trainer():
     assert getattr(trainer, 'show_progress_bar')
 
     with pytest.deprecated_call(match='v0.9.0'):
-        _ = Trainer(num_tpu_cores=8)
+        trainer = Trainer(num_tpu_cores=8)
+        assert trainer.tpu_cores == 8
 
 
 def test_tbd_remove_in_v0_9_0_module_imports():