def test_plot_snapshots_5(self):
dmd = DMD()
snapshots = [snap.reshape(20, 20) for snap in sample_data.T]
dmd.fit(X=snapshots)
dmd.plot_snapshots_2D(index_snap=2, filename='tmp.png')
self.addCleanup(os.remove, 'tmp.2.png')
def test_plot_snapshots_2(self):
dmd = DMD(svd_rank=-1)
dmd.fit(X=sample_data)
dmd.plot_snapshots_2D((1, 2, 5), x=np.arange(20), y=np.arange(20))
plt.close()
def test_plot_snapshots_4(self):
dmd = DMD()
snapshots = [snap.reshape(20, 20) for snap in sample_data.T]
dmd.fit(X=snapshots)
dmd.plot_snapshots_2D(index_snap=2)
plt.close()
def test_plot_snapshots_1(self):
dmd = DMD()
dmd.fit(X=sample_data)
with self.assertRaises(ValueError):
dmd.plot_snapshots_2D()
