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))