[tests.classify.test_montgomery] Simplify model tests

src/mednet/models/classify/densenet.py

@@ -70,7 +70,7 @@ class Densenet(Model):
         num_classes: int = 1,
     ):
         super().__init__(
-            name="densenet-121",
+            name="densenet",
             loss_type=loss_type,
             loss_arguments=loss_arguments,
             optimizer_type=optimizer_type,

tests/classify/test_montgomery.py

@@ -164,22 +164,12 @@ def test_model_transforms_image_quality(
     datamodule.model_transforms = model.model_transforms
     datamodule.setup("predict")
 
-    # Densenet's model.name is "densenet-121" and does not correspond to its
-    # module name.
-    if model_name == "densenet":
-        reference_histogram_file = (
-            datadir
-            / "histograms"
-            / "models"
-            / "histograms_densenet-121_montgomery_default.json"
-        )
-    else:
-        reference_histogram_file = (
-            datadir
-            / "histograms"
-            / "models"
-            / f"histograms_{model_name}_montgomery_default.json"
-        )
+    reference_histogram_file = (
+        datadir
+        / "histograms"
+        / "models"
+        / f"histograms_{model_name}_montgomery_default.json"
+    )
 
     database_checkers.check_image_quality(
         datamodule,