wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-06-30 14:03:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

ENH: Add removal of elements in benchmark.

Browse Source
This commit is contained in:
Thomas Wiecki
2014-06-04 16:07:23 +02:00
parent 79c91897ba
commit e4c974f4b8
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_data_util.py
+10 -4
View File
@@ -60,8 +60,8 @@ class TestRollingPanel(unittest.TestCase):
tm.assert_panel_equal(result, expected.swapaxes(0, 1))
def f(option='clever', n=500, copy=False):
items = range(5)
def run_history_implementations(option='clever', n=500, copy=False):
items = range(15)
minor = range(20)
window = 100
periods = n
@@ -72,10 +72,16 @@ def f(option='clever', n=500, copy=False):
if option == 'clever':
rp = RollingPanel(window, items, minor, cap_multiple=2)
major_deque = deque()
dummy = pd.DataFrame(np.random.randn(len(items), len(minor)),
index=items, columns=minor)
for i in range(periods):
if len(minor) > 5:
minor = minor[:-1]
if len(items) > 5:
items = items[:-1]
dummy = pd.DataFrame(np.random.randn(len(items), len(minor)),
index=items, columns=minor)
frame = dummy * (1 + 0.001 * i)
date = dates[i]