def test_plot_xy_with_hue(self): """Testing plot_xy_with_hue.""" # Here is some sample data. The 'x2' data is slightly offset from 'x1' x1 = list(range(0, 100, 10)) x2 = list(range(1, 100, 10)) x = x1 + x2 #The y-values I generate here mimic the general shape of my actual data y1 = x1[::-1] y2 = [i+25 for i in x1[::-1]] y = y1 + y2 # Two levels of labels that will be applied to the data z1 = ["1"] * 10 z2 = ["2"] * 10 z = z1 + z2 # A pandas data frame from the above data import pandas as pd df = pd.DataFrame({'x': x, 'y': y, 'z': z}) from abipy.tools.plotting import plot_xy_with_hue if self.has_matplotlib(): assert plot_xy_with_hue(data=df, x="x", y="y", hue="z", ax=None, show=False) assert plot_xy_with_hue(data=df, x="x", y="y", hue="z", ax=None, show=False, color="red", marker="v") assert plot_xy_with_hue(data=df, x="x", y="y", hue="z", decimals=0, ax=None, show=False, color="red", marker="v") with self.assertRaises(ValueError): plot_xy_with_hue(data=df, x="foo", y="y", hue="bar", ax=None, show=False)
def test_plot_xy_with_hue(self): """Testing plot_xy_with_hue.""" # Here is some sample data. The 'x2' data is slightly offset from 'x1' x1 = list(range(0, 100, 10)) x2 = list(range(1, 100, 10)) x = x1 + x2 #The y-values I generate here mimic the general shape of my actual data y1 = x1[::-1] y2 = [i + 25 for i in x1[::-1]] y = y1 + y2 # Two levels of labels that will be applied to the data z1 = ["1"] * 10 z2 = ["2"] * 10 z = z1 + z2 # A pandas data frame from the above data import pandas as pd df = pd.DataFrame({'x': x, 'y': y, 'z': z}) from abipy.tools.plotting import plot_xy_with_hue if self.has_matplotlib(): assert plot_xy_with_hue(data=df, x="x", y="y", hue="z", ax=None, show=False) assert plot_xy_with_hue(data=df, x="x", y="y", hue="z", ax=None, show=False, color="red", marker="v") assert plot_xy_with_hue(data=df, x="x", y="y", hue="z", decimals=0, ax=None, show=False, color="red", marker="v") with self.assertRaises(ValueError): plot_xy_with_hue(data=df, x="foo", y="y", hue="bar", ax=None, show=False)
def plot_xy_with_hue(*args, **kwargs): return plot_xy_with_hue(*args, **kwargs)
def plot_xy_with_hue(*args, **kwargs): return plot_xy_with_hue(*args, **kwargs)