def METHOD_NAME(self, mock_compute_operation: ComputeOperations, mocker: MockFixture) -> None:
    compute = AmlCompute(
        name="name",
        tags={"key1": "value1", "key2": "value2"},
        min_instances=0,
        max_instances=10,
        idle_time_before_scale_down=100,
        identity=IdentityConfiguration(
            type="UserAssigned",
            user_assigned_identities=[
                ManagedIdentityConfiguration(
                    resource_id="/subscriptions/b17253fa-f327-42d6-9686-f3e553e24763/resourcegroups/MC_banibatch_bani-aks_eastus/providers/Microsoft.ManagedIdentity/userAssignedIdentities/omsagent-bani-aks"
                )
            ],
        ),
    )
    mock_compute_operation.begin_update(compute)
    mock_compute_operation._operation.begin_create_or_update.assert_called_once()