transforms: stencil-tensorize-z-dim accept stencil.field as function arg

[n-io]

A minor fix (plus test case) that allows the transformation to work on stencil.field passed in as function arguments.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 89.80%. Comparing base (f33372f) to head (080c50a).

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #2756   +/-   ##
=======================================
  Coverage   89.80%   89.80%           
=======================================
  Files         372      372           
  Lines       47805    47807    +2     
  Branches     7328     7330    +2     
=======================================
+ Hits        42930    42934    +4     
+ Misses       3737     3736    -1     
+ Partials     1138     1137    -1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[PapyChacal]

Yay 😀

[n-io]

Yay 😀

As you suggested, this is actually a lot neater (thanks!), but best to allow both since we already support the other.