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