def test_rose_bools(data_fractures_compilation):
"""
Test supplying a data file containing a list of fracture lengths and
azimuth as digitized from geological maps to the data argument (lengths are
stored in the second column, azimuths in the first, specify via columns).
Test bools.
"""
fig = Figure()
fig.rose(
data=data_fractures_compilation,
region=[0, 1, 0, 360],
sector=10,
columns=[1, 0],
diameter="10c",
frame=["x0.2g0.2", "y30g30", "+glightgray"],
color="red3",
pen="1p",
orientation=False,
norm=True,
vectors=True,
no_scale=True,
shift=False,
)
return fig
def test_rose_deprecate_columns_to_incols(data_fractures_compilation):
"""
Make sure that the old parameter "columns" is supported and it reports a
warning.
Modified from the test_rose_bools() test.
"""
fig = Figure()
with pytest.warns(expected_warning=FutureWarning) as record:
fig.rose(
data=data_fractures_compilation,
region=[0, 1, 0, 360],
sector=10,
columns=[1, 0],
diameter="10c",
frame=["x0.2g0.2", "y30g30", "+glightgray"],
color="red3",
pen="1p",
orientation=False,
norm=True,
vectors=True,
no_scale=True,
shift=False,
)
assert len(record) == 1 # check that only one warning was raised
return fig
def test_rose_data_file(data_fractures_compilation):
"""
Test supplying data from sample dataset.
"""
fig = Figure()
fig.rose(
data=data_fractures_compilation,
region=[0, 1, 0, 360],
sector=15,
diameter="5.5c",
color="blue",
frame=["x0.2g0.2", "y30g30", "+glightgray"],
pen="1p",
norm="",
scale=0.4,
)
return fig
def test_rose_2d_array_multiple(data):
"""
Test supplying a 2D numpy array containing a list of lengths and
directions.
"""
fig = Figure()
fig.rose(
data=data,
region=[0, 1, 0, 360],
sector=10,
diameter="5.5c",
color="blue",
frame=["x0.2g0.2", "y30g30", "+gmoccasin"],
pen="1p",
norm=True,
scale=0.4,
)
return fig
def test_rose_2d_array_single():
"""
Test supplying a 2D numpy array containing a single pair of lengths and
directions.
"""
data = np.array([[40, 60]])
fig = Figure()
fig.rose(
data=data,
region=[0, 1, 0, 360],
sector=10,
diameter="5.5c",
color="cyan",
frame=["x0.2g0.2", "y30g30", "+glightgray"],
pen="1p",
norm=True,
scale=0.4,
)
return fig
def test_rose_no_sectors(data_fractures_compilation):
"""
Test supplying the sample fractures compilation dataset to the data
parameter.
Plot data without defining a sector width, add a title and rename labels.
"""
fig = Figure()
fig.rose(
data=data_fractures_compilation,
region=[0, 500, 0, 360],
diameter="10c",
labels="180/0/90/270",
frame=["xg100", "yg45", "+t'Windrose diagram'"],
pen="1.5p,red3",
transparency=40,
scale=0.5,
)
return fig
def test_rose_plot_data_using_cpt(data):
"""
Test supplying a 2D numpy array containing a list of lengths and
directions.
Use a cmap to color sectors.
"""
fig = Figure()
fig.rose(
data=data,
region=[0, 1, 0, 360],
sector=15,
diameter="5.5c",
cmap="batlow",
frame=["x0.2g0.2", "y30g30", "+gdarkgray"],
pen="1p",
norm=True,
scale=0.4,
)
return fig
def test_rose_no_sectors(data_fractures_compilation):
"""
Test supplying a data file containing a list of fracture lengths and
azimuth as digitized from geological maps to the data argument (lengths are
stored in the second column, azimuths in the first, specify via columns).
Plot data without defining a sector width, add a title and rename labels.
"""
fig = Figure()
fig.rose(
data=data_fractures_compilation,
region=[0, 500, 0, 360],
columns="1,0",
diameter="10c",
labels="180/0/90/270",
frame=["xg100", "yg45", "+t'Windrose diagram'"],
pen="1.5p,red3",
transparency=40,
scale=0.5,
)
return fig
def test_rose_plot_with_transparency(data_fractures_compilation):
"""
Test supplying the sample fractures compilation dataset to the data
parameter.
Use transparency.
"""
fig = Figure()
fig.rose(
data=data_fractures_compilation,
region=[0, 1, 0, 360],
sector=15,
diameter="5.5c",
color="blue",
frame=["x0.2g0.2", "y30g30", "+glightgray"],
pen="1p",
norm=True,
scale=0.4,
transparency=50,
)
return fig
def test_rose_plot_with_transparency(data_fractures_compilation):
"""
Test supplying a data file containing a list of fracture lengths and
azimuth as digitized from geological maps to the data argument (lengths are
stored in the second column, azimuths in the first, specify via columns).
Use transparency.
"""
fig = Figure()
fig.rose(
data=data_fractures_compilation,
region=[0, 1, 0, 360],
sector=15,
diameter="5.5c",
color="blue",
frame=["x0.2g0.2", "y30g30", "+glightgray"],
columns=[1, 0],
pen="1p",
norm=True,
scale=0.4,
transparency=50,
)
return fig
def test_rose_bools(data_fractures_compilation):
"""
Test supplying the sample fractures compilation dataset to the data
parameter.
Test bools.
"""
fig = Figure()
fig.rose(
data=data_fractures_compilation,
region=[0, 1, 0, 360],
sector=10,
diameter="10c",
frame=["x0.2g0.2", "y30g30", "+glightgray"],
color="red3",
pen="1p",
orientation=False,
norm=True,
vectors=True,
no_scale=True,
shift=False,
)
return fig