def test_clean_mask(chip_mask): from hsviz import draw_func2 as df2 mask = chip_mask print('Cleaning') mask2 = clean_mask(mask, 0, 3, .020) mask3 = clean_mask(mask, 3, 0, .023) mask4 = clean_mask(mask, 3, 3, .025) mask5 = clean_mask(mask4, 2, 3, .025) mask6 = clean_mask(mask5, 1, 0, .025) mask7 = clean_mask(mask6, 1, 0, .025) mask8 = clean_mask(mask7, 1, 0, .025) mask9 = clean_mask(mask8, 1, 3, .025) print('Drawing') df2.imshow(mask, pnum=331) df2.imshow(mask2, pnum=332) df2.imshow(mask3, pnum=333) df2.imshow(mask4, pnum=334) df2.imshow(mask5, pnum=335) df2.imshow(mask6, pnum=336) df2.imshow(mask7, pnum=337) df2.imshow(mask8, pnum=338) df2.imshow(mask9, pnum=339) print('Updating') df2.update() print('Done')
def test_encounter(hs): exec(open('hotspotter/encounter.py').read()) encounter.rrr() dev_api.dev_reload.reload_all_modules() try: if USE_TESTCACHE: raise KeyError('use_testcache=False') ex2_cxs = helpers.load_testdata('ex2_cxs', uid=hs.get_db_name()) except KeyError: ex2_cxs = encounter.get_chip_encounters(hs) helpers.save_testdata('ex2_cxs', uid=hs.get_db_name()) cxs = ex2_cxs[-1] assert len(cxs) > 1 qcx2_res = encounter.intra_query_cxs(hs, cxs) # Make a graph between the chips graph_netx = encounter.make_chip_graph(qcx2_res) netx.write_dot(graph_netx, 'encounter_graph.dot') try: import graph_tool graph = graph_tool.Graph() except ImportError as ex: print(ex) #encounter.viz_chipgraph(hs, graph, fnum=20, with_images=False) #encounter.viz_chipgraph(hs, graph, fnum=20, with_images=True) df2.update()
def viz_localmax(signal1d): #signal1d = np.array(hist) from hsviz import draw_func2 as df2 signal1d = np.array(signal1d) maxpos = np.array(localmax(signal1d)) x_data = range(len(signal1d)) y_data = signal1d df2.figure('localmax vizualization') df2.plot(x_data, y_data) df2.plot(maxpos, signal1d[maxpos], 'ro') df2.update()
def im(img, fnum=0): from hsviz import draw_func2 as df2 df2.imshow(img, fnum=fnum) df2.update()