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

PERF: Speed up event utils test case.

Browse Source 
Just take a sample of all 5000 permutations of 7 dates.
This commit is contained in:
Scott Sanderson
2016-07-14 17:38:58 -04:00
parent e22108b7ef
commit 67f76e4d67
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/pipeline/test_events.py
+7 -2
View File
@@ -483,8 +483,13 @@ class EventLoaderUtilsTestCase(ZiplineTestCase):
moscow_dates = pd.to_datetime(moscow_boundary_dates + mixed_tz_dates,
utc=True).tz_localize(None)
combos = list(map(np.array, itertools.permutations(np.arange(len(
boundary_dates + mixed_tz_dates)))))
all_combos = list(map(np.array, itertools.permutations(np.arange(len(
boundary_dates + mixed_tz_dates)
))))
# len(permutations(7)) is about 5000, which makes this take too long.
# Sampling down to 50-ish permutations still gives is good coverage of the
# different interleavings.
combos = all_combos[::100]
expected_us = pd.Series(
[pd.Timestamp('2013-01-04'),