def METHOD_NAME():
    model = ModelForNameTest()
    act_bn = get_all_modules_by_type(model, "BatchNorm2d")
    act_bn = OrderedDict((str(k), v) for k, v in act_bn.items())
    ref_bn = {
        "ModelForNameTest/BatchNorm2d[bn1]": model.bn1,
        "ModelForNameTest/BatchNorm2d[bn2]": model.bn2,
        "ModelForNameTest/BatchNorm2d[norm10]": model.norm10,
        "ModelForNameTest/BatchNorm2d[norm20]": model.norm20,
    }
    assert act_bn == ref_bn