That should work, yes.
This post from last week by @anthonylatsis contains an example that is very similar (doo is analogous to count and foo is analogous to distance(from:to:)).
I'll look into it. I think everything should get handled (tests are here, here, here and here).
Thank you.