test fixes

Author: awaelchli

tests/tests_pytorch/checkpointing/test_checkpoint_callback_frequency.py

@@ -54,8 +54,8 @@ def test_default_checkpoint_freq(save_mock, tmpdir, epochs: int, val_check_inter
 @pytest.mark.parametrize(
     ("k", "epochs", "val_check_interval", "expected"), [(1, 1, 1.0, 1), (2, 2, 1.0, 2), (2, 1, 0.25, 4), (2, 2, 0.3, 6)]
 )
-@pytest.mark.parametrize("save_last", [False, True])
-def test_top_k(save_mock, tmpdir, k: int, epochs: int, val_check_interval: float, expected: int, save_last: bool):
+@pytest.mark.parametrize("save_last", [False, True, "link"])
+def test_top_k(save_mock, tmpdir, k, epochs, val_check_interval, expected, save_last):
     class TestModel(BoringModel):
         def __init__(self):
             super().__init__()
@@ -79,8 +79,8 @@ def training_step(self, batch, batch_idx):
     )
     trainer.fit(model)
 
-    if save_last:
-        expected = expected
+    # save_last=True: last epochs are saved every step (so double the save calls)
+    expected = expected * 2 if save_last is True else expected
     assert save_mock.call_count == expected