diff --git a/harmonica/filters/_filters.py b/harmonica/filters/_filters.py index a561cef8d..daa708c97 100644 --- a/harmonica/filters/_filters.py +++ b/harmonica/filters/_filters.py @@ -472,7 +472,7 @@ def reduction_to_pole_kernel( magnetization_declination, ) # Set 0 wavenumber to 0 - da_filter.loc[dict(freq_northing=0, freq_easting=0)] = 0 + da_filter.loc[{dims[0]: 0, dims[1]: 0}] = 0 return da_filter diff --git a/harmonica/tests/test_transformations.py b/harmonica/tests/test_transformations.py index d93b365bb..05f04d1fe 100644 --- a/harmonica/tests/test_transformations.py +++ b/harmonica/tests/test_transformations.py @@ -523,6 +523,16 @@ def test_upward_continuation(sample_g_z, sample_g_z_upward): xrt.assert_allclose(continuation, g_z_upward, atol=1e-8) +def test_reduction_to_pole(sample_potential): + """ + Test reduction_to_pole function with non-typical dim names + """ + renamed_dims_grid = sample_potential.rename( + {"easting": "name_one", "northing": "name_two"} + ) + reduction_to_pole(renamed_dims_grid, 60, 45) + + class TestTotalGradientAmplitude: """ Test total_gradient_amplitude function