Add test assertion (#9309)

Author: carmocca

tests/trainer/loops/test_evaluation_loop.py

@@ -16,8 +16,9 @@
 import torch
 from torch.utils.data import DataLoader
 
-from pytorch_lightning import LightningModule, Trainer
+from pytorch_lightning import Trainer
 from pytorch_lightning.loops import EvaluationEpochLoop
+from pytorch_lightning.utilities.model_helpers import is_overridden
 from tests.helpers.boring_model import BoringModel, RandomDataset
 from tests.helpers.runif import RunIf
 
@@ -120,8 +121,8 @@ def on_advance_end(self):
             super().on_advance_end()
 
     model = TestModel()
-    # make sure this hook is not overridden
-    model.test_epoch_end = LightningModule.test_epoch_end
+    model.test_epoch_end = None
+    assert not is_overridden("test_epoch_end", model)
 
     trainer = Trainer(default_root_dir=tmpdir, fast_dev_run=3)
     trainer.test_loop.connect(TestLoop())