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