def METHOD_NAME(self, arg: str, mock_workspace_operation: WorkspaceOperations) -> None:
    mock_workspace_operation.list(scope=arg)
    if arg == "subscription":
        mock_workspace_operation._operation.list_by_subscription.assert_called_once()
    else:
        mock_workspace_operation._operation.list_by_resource_group.assert_called_once()