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)
