DedalusProject / dedalus

A flexible framework for solving PDEs with modern spectral methods.
http://dedalus-project.org/
GNU General Public License v3.0
489 stars 115 forks source link

Polar components on tensors #295

Closed lecoanet closed 1 month ago

lecoanet commented 2 months ago

Right now, the polar component operators (PolarRadialComponent and PolarAzimuthalComponent) assume they are acting on vectors. This PR allows the operators to act on tensors.

lecoanet commented 2 months ago

@kburns I had intended to add some tests to this, but it is pretty tricky to set up a test for this type of problem (non-square operator). So I think we might want to just merge this in directly. Thoughts?