mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-25 13:10:33 +08:00
MAINT: Remove unused function in test_rolling_panel.py.
This commit is contained in:
@@ -115,60 +115,3 @@ class TestRollingPanel(unittest.TestCase):
|
||||
|
||||
expected_minor.append(add_item)
|
||||
expected_items.append(add_item)
|
||||
|
||||
|
||||
def run_history_implementations(option='clever', n=500, change_fields=False,
|
||||
copy=False, n_items=15, n_minor=20,
|
||||
change_freq=5, window=100):
|
||||
items = range(n_items)
|
||||
minor = range(n_minor)
|
||||
periods = n
|
||||
|
||||
dates = pd.date_range('2000-01-01', periods=periods, tz='utc')
|
||||
frames = {}
|
||||
|
||||
if option == 'clever':
|
||||
rp = RollingPanel(window, items, minor, cap_multiple=2)
|
||||
major_deque = deque()
|
||||
|
||||
for i in range(periods):
|
||||
# Add a new and drop an field every change_freq iterations
|
||||
if change_fields and (i % change_freq) == 0:
|
||||
minor = minor[1:]
|
||||
minor.append(minor[-1] + 1)
|
||||
items = items[1:]
|
||||
items.append(items[-1] + 1)
|
||||
|
||||
dummy = pd.DataFrame(np.random.randn(len(items), len(minor)),
|
||||
index=items, columns=minor)
|
||||
|
||||
frame = dummy * (1 + 0.001 * i)
|
||||
date = dates[i]
|
||||
|
||||
rp.add_frame(date, frame)
|
||||
|
||||
frames[date] = frame
|
||||
major_deque.append(date)
|
||||
|
||||
if i >= window:
|
||||
del frames[major_deque.popleft()]
|
||||
|
||||
result = rp.get_current()
|
||||
if copy:
|
||||
result = result.copy()
|
||||
else:
|
||||
major_deque = deque()
|
||||
dummy = pd.DataFrame(np.random.randn(len(items), len(minor)),
|
||||
index=items, columns=minor)
|
||||
|
||||
for i in range(periods):
|
||||
frame = dummy * (1 + 0.001 * i)
|
||||
date = dates[i]
|
||||
frames[date] = frame
|
||||
major_deque.append(date)
|
||||
|
||||
if i >= window:
|
||||
del frames[major_deque.popleft()]
|
||||
|
||||
result = pd.Panel(frames, items=list(major_deque),
|
||||
major_axis=items, minor_axis=minor)
|
||||
|
||||
Reference in New Issue
Block a user