def test_dryrun_find_featuremap_dims(backbone: str, input_size: tuple[int, int]) -> None:
    """Use the function and check the expected output format."""
    layers = ["layer1", "layer2", "layer3"]
    model = TimmFeatureExtractor(backbone=backbone, layers=layers, pre_trained=True)
    mapping = dryrun_find_featuremap_dims(model, input_size, layers)
    for lay in layers:
        layer_mapping = mapping[lay]
        num_features = layer_mapping["num_features"]
        assert isinstance(num_features, int), f"{type(num_features)}"
        resolution = layer_mapping["resolution"]
        assert isinstance(resolution, tuple), f"{type(resolution)}"
        assert len(resolution) == len(input_size), f"{len(resolution)}, {len(input_size)}"
        assert all(isinstance(x, int) for x in resolution), f"{[type(x) for x in resolution]}"
Use the function and check the expected output format.
tests/unit/models/test_feature_extractor.py