def test_wrong_bin(self): max_level = 5 level = 2 dmd = MrDMD(max_level=max_level) dmd.fit(sample_data) with self.assertRaises(ValueError): dmd.partial_modes(level=level, node=2**level)
def test_partial_modes2(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) pmodes = dmd.partial_modes(level, 3) assert pmodes.shape == (sample_data.shape[0], rank)
def test_wrong_level(self): max_level = 5 dmd = MrDMD(max_level=max_level) dmd.fit(sample_data) with self.assertRaises(ValueError): dmd.partial_modes(max_level + 1)