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

Removes unused parameters in test trade info creation.

Browse Source 
This code hasn't used `random` in a while, removing the parameter
left as an artifact.
This commit is contained in:
Eddie Hebert
2013-01-26 09:43:22 -05:00
parent 629b6d6bd8
commit b3e8d56c02
1 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/sources/test_source.py
+6 -16
View File
@@ -17,7 +17,6 @@
A source to be used in testing.
"""
import random
import pytz
from itertools import cycle, ifilter, izip
@@ -42,29 +41,20 @@ def date_gen(start=datetime(2006, 6, 6, 12, tzinfo=pytz.utc),
return (start + (i * delta) for i in xrange(count))
def mock_prices(count, rand=False):
def mock_prices(count):
"""
Utility to generate a stream of mock prices. By default
cycles through values from 0.0 to 10.0, n times. Optional
flag to give random values between 0.0 and 10.0
cycles through values from 0.0 to 10.0, n times.
"""
if rand:
return (random.uniform(1.0, 10.0) for i in xrange(count))
else:
return (float(i % 10) + 1.0 for i in xrange(count))
return (float(i % 10) + 1.0 for i in xrange(count))
def mock_volumes(count, rand=False):
def mock_volumes(count):
"""
Utility to generate a set of volumes. By default cycles
through values from 100 to 1000, incrementing by 50. Optional
flag to give random values between 100 and 1000.
through values from 100 to 1000, incrementing by 50.
"""
if rand:
return (random.randrange(100, 1000) for i in xrange(count))
else:
return ((i * 50) % 900 + 100 for i in xrange(count))
return ((i * 50) % 900 + 100 for i in xrange(count))
class SpecificEquityTrades(object):