I'm not sure we can use `_cos` and `_sin` here--unless I'm misreading something about the code (which is entirely possible), those are computed using `i_kx`, and this is using `kx`, so they'll be equivalent if self.nx = self.ny but not otherwise.