def test_wrong_partial_reconstructed(self):
max_level = 5
level = 2
rank = 2
dmd = MrDMD(svd_rank=rank, max_level=max_level, max_cycles=2)
dmd.fit(X=sample_data)
with self.assertRaises(ValueError):
pdata = dmd.partial_reconstructed_data(max_level, 2)
def test_partial_reconstructed1(self):
max_level = 5
level = 2
rank = 2
dmd = MrDMD(svd_rank=rank, max_level=max_level, max_cycles=2)
dmd.fit(X=sample_data)
pdata = dmd.partial_reconstructed_data(level)
assert pdata.shape == sample_data.shape
def test_partial_reconstructed2(self):
max_level = 5
level = 2
rank = 2
dmd = MrDMD(svd_rank=rank, max_level=max_level, max_cycles=2)
dmd.fit(X=sample_data)
pdata = dmd.partial_reconstructed_data(level, 3)
assert pdata.shape == (sample_data.shape[0], old_div(
sample_data.shape[1], 2**level))