def METHOD_NAME(module, layers, param_name):
    module = module.to("cuda")
    analyzers = ModulePruningAnalyzer.analyze_layers(module, layers, param_name)
    assert len(analyzers) == len(layers)
    assert isinstance(analyzers[-1], ModulePruningAnalyzer)