def test_default(master=False, remove=True, show=False): name = osjoin(MASTER, 'default_master') if master else 'default' # Make the plot sub = df[(df.Substrate == 'Si') & (df['Target Wavelength'] == 450) & (df['Boost Level'] == 0.2) & (df['Temperature [C]'] == 25)] fcp.plot(df=sub, x='Voltage', y='I [A]', legend='Die', 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_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_marker_type_none(master=False, remove=True, show=False): name = osjoin(MASTER, 'marker_type_none_master') if master else 'marker_type_none' # Make the plot fcp.plot(df=df, x='Voltage', y='I [A]', legend=['Die', 'Substrate'], \ filter='Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25', markers=['o', None, '+', '*', 'B', None], 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_other_conf_int(master=False, remove=True, show=False): name = osjoin(MASTER, 'other_conf-int_master') if master else 'other_conf-int' # Make the plot fcp.plot( df, x='Voltage', y='I [A]', title='IV Data', lines=False, show=SHOW, filter= 'Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25', conf_int=0.95, 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_secondary_xy_shared_x(master=False, remove=True, show=False): name = osjoin( MASTER, 'secondary-xy_shared-x_master') if master else 'secondary-xy_shared-x' # Make the plot fcp.plot( df, x=['Voltage', 'I [A]'], y='Voltage', legend='Die', twin_y=True, show=SHOW, filter= 'Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25 & Die=="(-1,2)"', 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_wrap(master=False, remove=True, show=False): name = osjoin(MASTER, 'grid-plots-wrap_master') if master else 'grid-plots-wrap' # Make the plot fcp.plot(df, x='Voltage', y='I [A]', legend='Die', wrap=['Temperature [C]', 'Boost Level'], show=SHOW, ax_size=[225, 225], filter='Substrate=="Si" & Target Wavelength==450', label_rc_font_size=13, 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_violin_box_off(master=False, remove=True, show=False): name = osjoin(MASTER, 'violin_box_off_master') if master else 'violin_box_off' # Make the plot fcp.boxplot(df=df, y='Value', groups=['Batch', 'Sample'], show=SHOW, violin=True, violin_box_on=False, violin_markers=True, jitter=False, 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_range_lines(master=False, remove=True, show=False): name = osjoin(MASTER, 'range_lines_master') if master else 'range_lines' # Make the plot fcp.boxplot(df=df, y='Value', groups=['Batch', 'Sample'], show=SHOW, box_range_lines=False, ax_size=[300, 300], filename=name + '.png', inline=False, jitter=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_grid_wrap_y_no_share(master=False, remove=True, show=False): name = osjoin(MASTER, 'grid_y-no-share_master') if master else 'grid_y-no-share' # Make the plot df['Value*2'] = 2 * df['Value'] fcp.boxplot(df=df, y=['Value', 'Value*2'], groups=['Batch', 'Sample', 'Region'], wrap='y', show=SHOW, ax_size=[300, 300], share_y=False, filename=name + '.png', inline=False, jitter=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_mean_diamonds(master=False, remove=True, show=False): name = osjoin(MASTER, 'mean_diamonds_master') if master else 'mean_diamonds' # Make the plot fcp.boxplot(df=df, y='Value', groups=['Batch', 'Sample'], show=SHOW, mean_diamonds=True, conf_coeff=0.95, filename=name + '.png', inline=False, jitter=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_shared_no(master=False, remove=True, show=False): name = osjoin(MASTER, 'shared_no_master') if master else 'shared_no' # Make the plot fcp.plot(df, x='Voltage', y='I [A]', legend='Die', col='Boost Level', row='Temperature [C]', \ ax_size=[225, 225], filter='Substrate=="Si" & Target Wavelength==450', label_rc_font_size=14, xmin=[0, 0.1, 0.2, 0.3, 0.4], ymax=[1, 2, 3, 4, 5, 6], 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_shared_cols(master=False, remove=True, show=False): name = osjoin(MASTER, 'shared_columns_master') if master else 'shared_columns' # Make the plot sub = df[(df.Substrate == 'Si') & (df['Target Wavelength'] == 450)].copy() fcp.plot(df=sub, x='Voltage', y='I [A]', legend='Die', col='Boost Level', row='Temperature [C]', show=SHOW, ax_size=[225, 225], share_col=True, 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_boxplot_iqr(master=False, remove=True, show=False): name = osjoin(MASTER, 'boxplot_iqr_master') if master else 'boxplot_iqr' # Make the plot fcp.boxplot(df=df_box, y='Value', groups=['Batch', 'Sample'], filter='Batch==101', show=SHOW, ymin='1.5*iqr', ymax='1.5*iqr', filename=name + '.png', inline=False, jitter=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_xy_categorical(master=False, remove=True, show=False): name = osjoin(MASTER, 'xy_categorical_master') if master else 'xy_categorical' # Make the plot fcp.plot( df, x='Die', y='I [A]', show=SHOW, filter= 'Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25 & Voltage==1.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_group_single(master=False, remove=True, show=False): name = osjoin(MASTER, 'group_single_master') if master else 'group_single' # Make the plot fcp.boxplot(df=df, y='Value', groups='Batch', show=SHOW, filename=name + '.png', inline=False, jitter=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_xy_ts(master=False, remove=True, show=False): name = osjoin(MASTER, 'xy_ts_master') if master else 'xy_ts' # Make the plot fcp.plot(ts, x='Date', y='Happiness Quotient', markers=False, ax_size=[1000, 250], 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_filled(master=False, remove=True, show=False): name = osjoin(MASTER, 'filled_master') if master else 'filled' # Make the plot fcp.contour(df=df, x='X', y='Y', z='Value', row='Batch', col='Experiment', filled=True, cbar=True, xmin=-3, xmax=3, ymin=-3, ymax=3, ax_size=[250,250], show=SHOW, label_rc_font_size=12, levels=40, 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) 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_multiple_xy_both(master=False, remove=True, show=False): name = osjoin(MASTER, 'multiple-xy_both_master') if master else 'multiple-xy_both' # Make the plot fcp.plot( df, x=['Boost Level', 'I [A]'], y=['Voltage', 'Temperature [C]'], legend='Die', show=SHOW, filter= 'Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25', 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_legend_multiple_xy(master=False, remove=True, show=False): name = osjoin( MASTER, 'legend_multiple_xy_master') if master else 'legend_multiple_xy' # Make the plot fcp.plot( df=df1, x='Voltage', y=['I [A]', 'Voltage'], lines=False, filter='Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2', 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_other_stat_good_mult(master=False, remove=True, show=False): name = osjoin(MASTER, 'other_stat-lines-good-multipe-y_master' ) if master else 'other_stat-lines-good-multipe-y' # Make the plot fcp.plot( df, x='Voltage', y=['Boost Level', 'I [A]'], show=SHOW, legend=True, stat='median', filter= 'Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25', 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_groups_enabled2(master=False, remove=True, show=False): name = osjoin(MASTER, 'groups_enabled2_master') if master else 'groups_enabled2' # Make the plot fcp.plot( df=df1, x='Voltage', y='I [A]', groups=['Die', 'Temperature [C]'], filter='Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2', 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_groups_boxplot(master=False, remove=True, show=False): name = osjoin(MASTER, 'groups_boxplot_master') if master else 'groups_boxplot' # Make the plot df_box = pd.read_csv( osjoin(os.path.dirname(fcp.__file__), 'tests', 'fake_data_box.csv')) fcp.boxplot(df=df_box, y='Value', groups=['Batch', 'Sample'], legend='Region', filename=name + '.png', inline=False, jitter=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_groups_row_col_x(master=False, remove=True, show=False): name = osjoin(MASTER, 'groups_row_col_x_master') if master else 'groups_row_col_x' # Make the plot fcp.plot( df=df1, x=['Voltage', 'I [A]'], y='Voltage', legend='Die', row='Boost Level', col='x', ax_size=[225, 225], filter='Substrate=="Si" & Target Wavelength==450 & Temperature [C]==75', label_rc_font_size=14, 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_hist_color(master=False, remove=True, show=False): name = osjoin(MASTER, 'hist_color_master') if master else 'hist_color' # Make the plot df_hist = pd.read_csv( osjoin(os.path.dirname(fcp.__file__), 'tests', 'fake_data_box.csv')) fcp.hist(df=df_hist, x='Value', show=SHOW, legend='Region', hist_fill_alpha=1, colors=['#FF0000', '#00FF11'], 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_groups_wrap_xy(master=False, remove=True, show=False): name = osjoin(MASTER, 'groups_wrap_xy_master') if master else 'groups_wrap_xy' # Make the plot fcp.plot(df=df1, x='Voltage', y=['I Set', 'I [A]'], legend='Die', wrap='y', groups=['Boost Level', 'Temperature [C]'], ax_size=[325, 325], filter='Substrate=="Si" & Target Wavelength==450', 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_theme_gray(master=False, remove=True, show=False): name = osjoin(MASTER, 'theme_gray_master') if master else 'theme_gray' # Make the plot fcp.set_theme('gray') fcp.plot(df=df, x='Voltage', y='I [A]', legend=['Die', 'Substrate'], filter= 'Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25', 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_primary_explicit(master=False, remove=True, show=False): name = osjoin(MASTER, 'primary_explicit_master') if master else 'primary_explicit' # Make the plot fcp.plot( df=df, x='Voltage', y='I [A]', legend='Die', show=SHOW, filter= 'Substrate=="Si" & Target Wavelength==450 & Boost Level==0.2 & Temperature [C]==25', xmax=1.2, auto_scale=False, 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