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)