def test_registry_plot(): class TestPlot(object): type = 'test' pass gwsumm_plot.register_plot(TestPlot) with pytest.raises(ValueError): gwsumm_plot.register_plot(TestPlot) gwsumm_plot.register_plot(TestPlot, force=True) assert gwsumm_plot.get_plot('test') is TestPlot gwsumm_plot.register_plot(TestPlot, name='test-with-name') assert gwsumm_plot.get_plot('test-with-name') is TestPlot
def test_registry_plot(): class TestPlot(object): type = 'test' pass gwsumm_plot.register_plot(TestPlot) with pytest.raises(ValueError): gwsumm_plot.register_plot(TestPlot) gwsumm_plot.register_plot(TestPlot, force=True) assert gwsumm_plot.get_plot('test') is TestPlot gwsumm_plot.register_plot(TestPlot, name='test-with-name') assert gwsumm_plot.get_plot('test-with-name') is TestPlot
def test_registry(self): # test get p = plot.get_plot(None) self.assertIs(p, plot.SummaryPlot) p = plot.get_plot('\'data\'') self.assertIs(p, plot.DataPlot) self.assertRaises(ValueError, plot.get_plot, 'dfskaewf') # test register class TestPlot(object): type = 'test' pass plot.register_plot(TestPlot) self.assertRaises(ValueError, plot.register_plot, TestPlot) plot.register_plot(TestPlot, force=True) self.assertIs(plot.get_plot('test'), TestPlot) plot.register_plot(TestPlot, name='test-with-name') self.assertIs(plot.get_plot('test-with-name'), TestPlot)
def test_registry(self): # test get p = plot.get_plot(None) self.assertIs(p, plot.SummaryPlot) p = plot.get_plot('\'data\'') self.assertIs(p, plot.DataPlot) self.assertRaises(ValueError, plot.get_plot, 'dfskaewf') # test register class TestPlot(object): type = 'test' pass plot.register_plot(TestPlot) self.assertRaises(ValueError, plot.register_plot, TestPlot) plot.register_plot(TestPlot, force=True) self.assertIs(plot.get_plot('test'), TestPlot) plot.register_plot(TestPlot, name='test-with-name') self.assertIs(plot.get_plot('test-with-name'), TestPlot)
def setup_class(cls): cls.PLOT = gwsumm_plot.get_plot(cls.TYPE)
def test_get_plot(name, plot): assert gwsumm_plot.get_plot(name) is plot with pytest.raises(ValueError): gwsumm_plot.get_plot('dfskaewf')
def init_plots(self, plotdir=os.curdir): """Configure the default list of plots for this tab This method configures a veto-trigger glitchgram, histograms of before/after SNR and frequency/template duration, before and after glitchgrams, and a segment plot. This method is a mess, and should be re-written in a better way. """ if self.intersection: label = 'Intersection' else: label = 'Union' self.layout = [ 1, ] before = get_channel(str(self.channel)) for state in self.states: if self.channel: after = get_channel( veto_tag(before, self.metaflag, mode='after')) vetoed = get_channel( veto_tag(before, self.metaflag, mode='vetoed')) # -- configure trigger plots params = etg.get_etg_parameters(self.etg) glitchgramargs = { 'etg': self.etg, 'x': 'time', 'y': params['frequency'], 'logy': params.get('frequency-log', True), 'ylabel': get_column_label(params['frequency']), 'edgecolor': 'none', 'legend-scatterpoints': 1, 'legend-borderaxespad': 0, 'legend-bbox_to_anchor': (1.01, 1), 'legend-loc': 'upper left', } # plot before/after glitchgram self.plots.append( get_plot('triggers')([after, vetoed], self.start, self.end, state=state, title='Impact of %s (%s)' % (label_to_latex(self.name), self.etg), outdir=plotdir, labels=['_', 'Vetoed'], colors=['lightblue', 'red'], **glitchgramargs)) # plot histograms statistics = ['snr'] if params['det'] != params['snr']: statistics.append('det') self.layout.append(len(statistics) + 1) for column in statistics + ['frequency']: self.plots.append( get_plot('trigger-histogram')( [before, after], self.start, self.end, state=state, column=params[column], etg=self.etg, outdir=plotdir, title='Impact of %s (%s)' % (label_to_latex(self.name), self.etg), labels=['Before', 'After'], xlabel=params.get('%s-label' % column, get_column_label( params[column])), color=['red', (0.2, 0.8, 0.2)], logx=params.get('%s-log' % column, True), logy=True, histtype='stepfilled', alpha=0.6, weights=1 / float(abs(self.span)), bins=100, ybound=1 / float(abs(self.span)) * 0.5, **{ 'legend-borderaxespad': 0, 'legend-bbox_to_anchor': (1.01, 1), 'legend-loc': 'upper left' })) # plot triggers before and after for stat in statistics: column = params[stat] glitchgramargs.update({ 'color': column, 'clim': params.get('%s-limits' % stat, [3, 100]), 'logcolor': params.get('%s-log' % stat, True), 'colorlabel': params.get('%s-label' % stat, get_column_label(column)), }) self.plots.append( get_plot('triggers')([before], self.start, self.end, state=state, outdir=plotdir, **glitchgramargs)) self.plots.append( get_plot('triggers')( [after], self.start, self.end, state=state, title='After %s (%s)' % (label_to_latex(self.name), self.etg), outdir=plotdir, **glitchgramargs)) self.layout.append(2) # -- configure segment plot segargs = { 'state': state, 'known': { 'alpha': 0.1, 'facecolor': 'lightgray' }, 'color': 'red', } if len(self.flags) == 1: sp = get_plot('segments')(self.flags, self.start, self.end, outdir=plotdir, labels=self.labels, **segargs) else: sp = get_plot('segments')( [self.metaflag] + self.flags, self.start, self.end, labels=([self.intersection and 'Intersection' or 'Union'] + self.labels), outdir=plotdir, **segargs) self.plots.append(sp) self.layout.append(1)
def init_plots(self, plotdir=os.curdir): """Configure the default list of plots for this tab This method configures a veto-trigger glitchgram, histograms of before/after SNR and frequency/template duration, before and after glitchgrams, and a segment plot. This method is a mess, and should be re-written in a better way. """ if self.intersection: label = 'Intersection' else: label = 'Union' etgstr = self.etg.replace('_', r'\\_') self.set_layout([1,]) before = get_channel(str(self.channel)) for state in self.states: if self.channel: after = get_channel(veto_tag(before, self.metaflag, mode='after')) vetoed = get_channel(veto_tag(before, self.metaflag, mode='vetoed')) # -- configure trigger plots params = etg.get_etg_parameters(self.etg, IFO=before.ifo) glitchgramargs = { 'etg': self.etg, 'x': params['time'], 'y': params['frequency'], 'logy': params.get('frequency-log', True), 'ylabel': params.get('frequency-label', get_column_label(params['frequency'])), 'edgecolor': 'none', 'legend-scatterpoints': 1, 'legend-borderaxespad': 0, 'legend-bbox_to_anchor': (1.01, 1), 'legend-loc': 'upper left', } # plot before/after glitchgram self.plots.append(get_plot('triggers')( [after, vetoed], self.start, self.end, state=state, title='Impact of %s (%s)' % ( label_to_latex(self.name), etgstr), outdir=plotdir, labels=['_', 'Vetoed'], colors=['lightblue', 'red'], **glitchgramargs)) # plot histograms statistics = ['snr'] if params['det'] != params['snr']: statistics.append('det') self.layout.append(len(statistics) + 1) for column in statistics + ['frequency']: self.plots.append(get_plot('trigger-histogram')( [before, after], self.start, self.end, state=state, column=params[column], etg=self.etg, outdir=plotdir, title='Impact of %s (%s)' % ( label_to_latex(self.name), etgstr), labels=['Before', 'After'], xlabel=params.get('%s-label' % column, get_column_label(params[column])), color=['red', (0.2, 0.8, 0.2)], logx=params.get('%s-log' % column, True), logy=True, histtype='stepfilled', alpha=0.6, weights=1/float(abs(self.span)), bins=100, ybound=1/float(abs(self.span)) * 0.5, **{ 'legend-borderaxespad': 0, 'legend-bbox_to_anchor': (1.01, 1), 'legend-loc': 'upper left'} )) # plot triggers before and after for stat in statistics: column = params[stat] glitchgramargs.update({ 'color': column, 'clim': params.get('%s-limits' % stat, [3, 100]), 'logcolor': params.get('%s-log' % stat, True), 'colorlabel': params.get('%s-label' % stat, get_column_label(column)), }) self.plots.append(get_plot('triggers')( [before], self.start, self.end, state=state, outdir=plotdir, **glitchgramargs)) self.plots.append(get_plot('triggers')( [after], self.start, self.end, state=state, title='After %s (%s)' % ( label_to_latex(self.name), self.etg), outdir=plotdir, **glitchgramargs)) self.layout.append(2) # -- configure segment plot segargs = { 'state': state, 'known': {'alpha': 0.1, 'facecolor': 'lightgray'}, 'color': 'red', } if len(self.flags) == 1: sp = get_plot('segments')(self.flags, self.start, self.end, outdir=plotdir, labels=self.labels, **segargs) else: sp = get_plot('segments')( [self.metaflag] + self.flags, self.start, self.end, labels=([self.intersection and 'Intersection' or 'Union'] + self.labels), outdir=plotdir, **segargs) self.plots.append(sp) self.layout.append(1)
def init_plots(self, plotdir=os.curdir): """Configure the default list of plots for this tab This method configures a veto-trigger glitchgram, histograms of before/after SNR and frequency/template duration, before and after glitchgrams, and a segment plot. This method is a mess, and should be re-written in a better way. """ namestr = self.title.split('/')[0] before = get_channel(str(self.channel)) for state in self.states: # -- configure segment plot segargs = { 'state': state, 'known': { 'alpha': 0.1, 'facecolor': 'lightgray' }, 'color': 'red', } if len(self.flags) == 1: sp = get_plot('segments')(self.flags, self.start, self.end, outdir=plotdir, labels=self.labels, title='Veto segments: %s' % (texify(namestr)), **segargs) else: sp = get_plot('segments')( [self.metaflag] + self.flags, self.start, self.end, labels=([self.intersection and 'Intersection' or 'Union'] + self.labels), outdir=plotdir, title='Veto segments: %s' % texify(namestr), **segargs) self.plots.append(sp) if self.channel: self.set_layout([ 2, ]) after = get_channel( veto_tag(before, self.metaflag, mode='after')) vetoed = get_channel( veto_tag(before, self.metaflag, mode='vetoed')) # -- configure trigger plots params = etg.get_etg_parameters(self.etg, IFO=before.ifo) glitchgramargs = { 'etg': self.etg, 'x': params['time'], 'y': params['frequency'], 'yscale': params.get('frequency-scale', 'log'), 'ylabel': params.get('frequency-label', get_column_label(params['frequency'])), 'ylim': params.get('ylim', [10, 5000]), 'edgecolor': 'none', 'legend-scatterpoints': 1, 'legend-borderaxespad': 0, 'legend-bbox_to_anchor': (1, 1), 'legend-loc': 'upper left', 'legend-frameon': False, } glitchgramargs.update(self.glitchgramargs) # plot before/after glitchgram self.plots.append( get_plot('triggers')([after, vetoed], self.start, self.end, state=state, title='Impact of %s (%s)' % (texify(namestr), self.etg), outdir=plotdir, labels=['After', 'Vetoed'], **glitchgramargs)) # plot histograms if params['det'] != params['snr']: statistics = ['snr', 'det'] xlims = [(5, 16384), (6, 15), (8, 8192)] else: statistics = ['snr'] xlims = [(5, 16384), (8, 8192)] self.layout.append(len(statistics) + 1) for column, xlim in zip(statistics + ['frequency'], xlims): self.plots.append( get_plot('trigger-histogram')( [before, after], self.start, self.end, state=state, column=params[column], etg=self.etg, outdir=plotdir, filterstr=self.filterstr, title='Impact of %s (%s)' % (texify(namestr), self.etg), labels=['Before', 'After'], xlim=xlim, xlabel=params.get('%s-label' % column, get_column_label( params[column])), color=['#ffa07a', '#1f77b4'], alpha=1, xscale=params.get('%s-scale' % column, 'log'), yscale='log', histtype='stepfilled', weights=1 / float(abs(self.span)), bins=100, ylim=(1 / float(abs(self.span)) * 0.5, 1), **{ 'legend-borderaxespad': 0, 'legend-bbox_to_anchor': (1., 1.), 'legend-loc': 'upper left', 'legend-frameon': False, })) # plot triggers before and after for stat in statistics: column = params[stat] glitchgramargs.update({ 'color': column, 'clim': params.get('%s-limits' % stat, [3, 100]), 'logcolor': params.get('%s-log' % stat, True), 'colorlabel': params.get('%s-label' % stat, get_column_label(column)), }) self.plots.append( get_plot('triggers')([before], self.start, self.end, state=state, outdir=plotdir, filterstr=self.filterstr, **glitchgramargs)) self.plots.append( get_plot('triggers')( [after], self.start, self.end, state=state, title='%s after %s (%s)' % (texify( str(self.channel)), texify(namestr), self.etg), outdir=plotdir, filterstr=self.filterstr, **glitchgramargs)) self.layout.append(2) else: self.set_layout([ 1, ])
def setup_class(cls): cls.PLOT = gwsumm_plot.get_plot(cls.TYPE)
def test_get_plot(name, plot): assert gwsumm_plot.get_plot(name) is plot with pytest.raises(ValueError): gwsumm_plot.get_plot('dfskaewf')