def test_wrong_plot_eig1(self): dmd = MrDMD(svd_rank=-1, max_level=7, max_cycles=1) dmd.fit(X=sample_data) with self.assertRaises(ValueError): dmd.plot_eigs( show_axes=True, show_unit_circle=True, figsize=(8, 8), level=7 )
def test_plot_eig3(self):
dmd = MrDMD(svd_rank=-1, max_level=7, max_cycles=1)
dmd.fit(X=sample_data)
dmd.plot_eigs(show_axes=False, show_unit_circle=False, level=1, node=0)
plt.close()
def test_plot_eig2(self):
dmd = MrDMD(svd_rank=-1, max_level=7, max_cycles=1)
dmd.fit(X=sample_data)
dmd.plot_eigs(show_axes=True, show_unit_circle=False, title='Title')
plt.close()
def test_plot_eig1(self):
dmd = MrDMD(svd_rank=-1, max_level=7, max_cycles=1)
dmd.fit(X=sample_data)
dmd.plot_eigs(show_axes=True, show_unit_circle=True, figsize=(8, 8))
plt.close()
def test_wrong_plot_eig2(self):
dmd = MrDMD(svd_rank=1, max_level=7, max_cycles=1)
with self.assertRaises(ValueError):
dmd.plot_eigs()