def test_partial_time_interval3(self):
level = 4
dmd = MrDMD(svd_rank=1, max_level=level, max_cycles=2)
dmd.original_time = {'t0': 0, 'tend': 8, 'dt': 1}
with self.assertRaises(ValueError):
dmd.partial_time_interval(3, 8)
def test_partial_time_interval(self):
level = 4
dmd = MrDMD(svd_rank=1, max_level=level, max_cycles=2)
dmd.original_time = {'t0': 0, 'tend': 8, 'dt': 1}
ans = {'t0': 6.0, 'tend': 7.0, 'dt': 1.0}
assert dmd.partial_time_interval(3, 6) == ans