From e4c974f4b863ed17eec01dad7933cbd4665dfcd1 Mon Sep 17 00:00:00 2001 From: Thomas Wiecki Date: Wed, 4 Jun 2014 16:07:23 +0200 Subject: [PATCH] ENH: Add removal of elements in benchmark. --- tests/test_data_util.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tests/test_data_util.py b/tests/test_data_util.py index ded2c8ca..7e62185f 100644 --- a/tests/test_data_util.py +++ b/tests/test_data_util.py @@ -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]