def test_wiggle_deprecate_columns_to_incols():
"""
Make sure that the old parameter "columns" is supported and it reports a
warning.
Modified from the test_wiggle() test.
"""
# put data into numpy array and swap x and y columns
# as the use of the 'columns' parameter will reverse this action
x = np.arange(-2, 2, 0.02)
y = np.zeros(x.size)
z = np.cos(2 * np.pi * x)
data = np.array([y, x, z]).T
fig = Figure()
with pytest.warns(expected_warning=FutureWarning) as record:
fig.wiggle(
data,
region=[-4, 4, -1, 1],
projection="X8c",
columns=[1, 0, 2],
scale="0.5c",
color=["red+p", "gray+n"],
pen="1.0p",
track="0.5p",
position="jRM+w2+lnT",
)
assert len(record) == 1 # check that only one warning was raised
return fig
def test_wiggle():
"""
Plot the z=f(x,y) anomalies along tracks.
"""
x = np.arange(-2, 2, 0.02)
y = np.zeros(x.size)
z = np.cos(2 * np.pi * x)
fig = Figure()
fig.wiggle(
region=[-4, 4, -1, 1],
projection="X8c",
x=x,
y=y,
z=z,
scale="0.5c",
color=["red+p", "gray+n"],
pen="1.0p",
track="0.5p",
position="jRM+w2+lnT",
)
return fig
