def test_line():
"""
Function tests line plots created by pyedahelper.fast_plot function.
"""
a = pyedahelper.fast_plot(df=df, x="col_int", y="col_flt",
plot_type="line")
assert a.height == 600
assert a.width == 900
assert a.mark == "line"
assert "col_int" in a.encoding['x']['shorthand']
assert "col_flt" in a.encoding['y']['shorthand']
b = pyedahelper.fast_plot(df=df, x="col_date", y="col_chr",
plot_type="line")
assert b.height == 600
assert b.width == 900
assert b.mark == "line"
assert "col_date" in b.encoding['x']['shorthand']
assert "col_chr" in b.encoding['y']['shorthand']
# raise error if y is date
with pytest.raises(Exception):
pyedahelper.fast_plot(df=df, x="col_int", y="col_date",
plot_type="line")
def test_bar():
"""
Function tests bar charts created by pyedahelper.fast_plot function.
"""
a = pyedahelper.fast_plot(df=df, x="col_int", y="col_flt", plot_type="bar")
assert a.height == 600
assert a.width == 900
assert a.mark == "bar"
assert "col_int" in a.encoding['x']['shorthand']
assert "col_flt" in a.encoding['y']['shorthand']
# ensure Exception gets raised when both columns are non-numeric
with pytest.raises(Exception):
pyedahelper.fast_plot(df=df,
x="col_date",
y="col_date",
plot_type="bar")
with pytest.raises(Exception):
pyedahelper.fast_plot(df=df, x="col_chr", y="col_chr", plot_type="bar")
with pytest.raises(Exception):
pyedahelper.fast_plot(df=df,
x="col_date",
y="col_chr",
plot_type="bar")
with pytest.raises(Exception):
pyedahelper.fast_plot(df=df,
x="col_chr",
y="col_date",
plot_type="bar")
def test_scatter():
"""
Function tests scatter plots created by pyedahelper.fast_plot function.
"""
a = pyedahelper.fast_plot(df=df, x="col_int", y="col_flt",
plot_type="scatter")
assert a.height == 600
assert a.width == 900
assert a.mark == "point"
assert "col_int" in a.encoding['x']['shorthand']
assert "col_flt" in a.encoding['y']['shorthand']
# raise error if y is date
with pytest.raises(Exception, match="Y column cannot be a date type!"):
pyedahelper.fast_plot(df=df, x="col_int", y="col_date",
plot_type="scatter")
def test_inputs():
"""
Function tests inputs for pyedahelper.fast_plot function.
"""
# data input isn't data frame
with pytest.raises(AssertionError,
match="Data must be in pandas Data Frame!"):
pyedahelper.fast_plot(df=[1, 2],
x="col_date",
y="col_int",
plot_type="line")
# x input isn't string
with pytest.raises(AssertionError,
match="x column name must be a string!"):
pyedahelper.fast_plot(df=df,
x=["col_date"],
y="col_int",
plot_type="line")
# y input isn't string
with pytest.raises(AssertionError,
match="y column name must be a string!"):
pyedahelper.fast_plot(df=df,
x="col_date",
y=["col_int"],
plot_type="line")
# x input is not a column in data frame
with pytest.raises(AssertionError,
match="x column name is "
"not a column in data frame entered!"):
pyedahelper.fast_plot(df=df,
x="asdfjsf",
y="col_int",
plot_type="line")
# y input is not a column in data frame
with pytest.raises(AssertionError,
match="y column name is "
"not a column in data frame entered!"):
pyedahelper.fast_plot(df=df,
x="col_date",
y="slkfjas",
plot_type="line")
# plot type not bar/line/scatter
with pytest.raises(AssertionError,
match='plot_type must be either:'
' "scatter", "line", or "bar"'):
pyedahelper.fast_plot(df=df,
x="col_date",
y="col_int",
plot_type="slfjalskf")
# x column should not be all null
with pytest.raises(AssertionError, match="x Column must not be all Null!"):
pyedahelper.fast_plot(df=df,
x="col_nan",
y="col_int",
plot_type="line")
# x column should not be all null
with pytest.raises(AssertionError, match="y Column must not be all Null!"):
pyedahelper.fast_plot(df=df,
x="col_int",
y="col_nan",
plot_type="line")