def test_indices(self): def test(G): G.plot(backend='matplotlib', indices=False) G.plot(backend='matplotlib', indices=True) # Test for 2D and 3D graphs. G = graphs.Ring(10) test(G) G = graphs.Torus(Nv=5) test(G)
def test_highlight(self): def test(G): s = np.arange(G.N) G.plot(s, backend='matplotlib', highlight=0) G.plot(s, backend='matplotlib', highlight=[0]) G.plot(s, backend='matplotlib', highlight=[0, 1]) # Test for 1, 2, and 3D graphs. G = graphs.Ring(10) test(G) G.set_coordinates('line1D') test(G) G = graphs.Torus(Nv=5) test(G)
def test_torus(self): graphs.Torus()
def test_Torus(): G = graphs.Torus()
g.plot(ax=ax) _ = ax.set_title('Filter bank of Meyer wavelets') DELTA = 255 s = g.localize(DELTA) fig = plt.figure(figsize=(10, 2.5)) for i in range(4): ax = fig.add_subplot(1, 4, i + 1, projection='3d') G.plot_signal(s[:, i], ax=ax) _ = ax.set_title('Wavelet {}'.format(i + 1)) ax.set_axis_off() fig.tight_layout() plt.show() # Mexican hat wavelets on a torus G = graphs.Torus() G.estimate_lmax() g = filters.MexicanHat(G, Nf=6) fig, ax = plt.subplots(figsize=(10, 5)) g.plot(ax=ax) _ = ax.set_title('Filter bank of Mexican Hat wavelets') DELTA = 255 s = g.localize(DELTA) fig = plt.figure(figsize=(10, 2.5)) for i in range(4): ax = fig.add_subplot(1, 4, i + 1, projection='3d') G.plot_signal(s[:, i], ax=ax) _ = ax.set_title('Wavelet {}'.format(i + 1)) ax.set_axis_off() fig.tight_layout()
def test_Torus(): G = graphs.Torus() needed_attributes_testing(G)