def test_cat_label(master=False, remove=True, show=False): name = osjoin(MASTER, 'cat_label_master') if master else 'cat_label' # Make the plot fcp.heatmap(df=df, x='Category', y='Player', z='Average', cbar=True, data_labels=True, heatmap_font_color='#aaaaaa', show=SHOW, tick_labels_major_y_edge_width=0, ws_ticks_ax=5, filename=name + '.png', inline=False) # Compare with master if master: return elif show: os.startfile(osjoin(MASTER, name + '_master.png')) os.startfile(name + '.png') compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png'), show=True) else: compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png')) if remove: os.remove(name + '.png') assert not compare
def test_cat_no_label(master=False, remove=True, show=False): name = osjoin(MASTER, 'cat_no_label_master') if master else 'cat_no_label' # Make the plot fcp.heatmap(df=df, x='Category', y='Player', z='Average', cbar=True, show=SHOW, filename=name + '.png', inline=False) # Compare with master if master: return elif show: os.startfile(osjoin(MASTER, name + '_master.png')) os.startfile(name + '.png') compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png'), show=True) else: compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png')) if remove: os.remove(name + '.png') assert not compare
def test_heatmap_stretched(master=False, remove=True, show=False): name = osjoin( MASTER, 'heatmap_stretched_master') if master else 'heatmap_stretched' # Make the plot uu = img.stack().mean() ss = img.stack().std() fcp.heatmap(img, cmap='inferno', cbar=True, ax_size=[600, 600], zmin=uu - 3 * ss, zmax=uu + 3 * ss, filename=name + '.png', inline=False) # Compare with master if master: return elif show: os.startfile(osjoin(MASTER, name + '_master.png')) os.startfile(name + '.png') compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png'), show=True) else: compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png')) if remove: os.remove(name + '.png') assert not compare
def test_heatmap_zoomed(master=False, remove=True, show=False): name = osjoin(MASTER, 'heatmap_zoomed_master') if master else 'heatmap_zoomed' # Make the plot fcp.heatmap(img, cmap='inferno', cbar=True, ax_size=[600, 600], xmin=700, xmax=1100, ymin=300, ymax=400, filename=name + '.png', inline=False) # Compare with master if master: return elif show: os.startfile(osjoin(MASTER, name + '_master.png')) os.startfile(name + '.png') compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png'), show=True) else: compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png')) if remove: os.remove(name + '.png') assert not compare
def test_cat_non_uniform(master=False, remove=True, show=False): name = osjoin(MASTER, 'cat_non-uniform_master') if master else 'cat_non-uniform' # Make the plot df2 = pd.read_csv( osjoin(os.path.dirname(fcp.__file__), 'tests', 'fake_data_contour.csv')) fcp.heatmap(df=df2, x='X', y='Y', z='Value', row='Batch', col='Experiment', cbar=True, show=SHOW, share_z=True, ax_size=[400, 400], data_labels=False, label_rc_font_size=12, filter='Batch==103', cmap='viridis', filename=name + '.png', inline=False) # Compare with master if master: return elif show: os.startfile(osjoin(MASTER, name + '_master.png')) os.startfile(name + '.png') compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png'), show=True) else: compare = utl.img_compare(name + '.png', osjoin(MASTER, name + '_master.png')) if remove: os.remove(name + '.png') assert not compare