def test_plot_snapshots_5(self): dmd = CDMD() 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 = CDMD(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 = CDMD() 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 = CDMD() dmd.fit(X=sample_data) with self.assertRaises(ValueError): dmd.plot_snapshots_2D()