def test_plot_modes_5(self):
dmd = HODMD()
snapshots = [snap.reshape(20, 20) for snap in sample_data.T]
dmd.fit(X=snapshots)
dmd.plot_modes_2D(index_mode=1, filename='tmp.png')
self.addCleanup(os.remove, 'tmp.1.png')
def test_plot_modes_2(self):
dmd = HODMD(svd_rank=-1)
dmd.fit(X=sample_data)
dmd.plot_modes_2D((1, 2, 5), x=np.arange(20), y=np.arange(20))
plt.close()
def test_plot_modes_4(self):
dmd = HODMD()
snapshots = [snap.reshape(20, 20) for snap in sample_data.T]
dmd.fit(X=snapshots)
dmd.plot_modes_2D(index_mode=1)
plt.close()
def test_plot_modes_1(self):
dmd = HODMD()
dmd.fit(X=sample_data)
with self.assertRaises(ValueError):
dmd.plot_modes_2D()