def test_time_window_bins(self): level = 4 dmd = MrDMD(svd_rank=1, max_level=level, max_cycles=2) dmd.original_time = {'t0': 0, 'tend': 9, 'dt': 1} comparison = dmd.time_window_bins(0, 9) == np.arange(2**level-1) assert comparison.all()
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
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_time_window_bins4(self): level = 3 dmd = MrDMD(svd_rank=1, max_level=level, max_cycles=2) dmd.original_time = {'t0': 0, 'tend': 4, 'dt': 1} comparison = dmd.time_window_bins(1, 3) == np.array([0, 1, 2, 4, 5]) assert comparison.all()