def METHOD_NAME():
    model = ModelForNameTest()
    act_bn = get_all_modules_by_type(model, ["ReLU", "AvgPool2d"])
    act_bn = OrderedDict((str(k), v) for k, v in act_bn.items())
    ref_bn = [
        "ModelForNameTest/AvgPool2d[avgpool]",
        "ModelForNameTest/Sequential[layer1]/ReLU[relu01]",
        "ModelForNameTest/Sequential[layer2]/ReLU[relu02]",
    ]
    assert list(act_bn.keys()) == ref_bn
    assert isinstance(act_bn, OrderedDict)