コード例 #1
0
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
コード例 #2
0
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