Added parameter type validation to record_variables

This commit is contained in:
Richard Frank
2013-02-11 18:07:05 -05:00
parent 59d76ce378
commit f914c53790
+5 -1
View File
@@ -283,9 +283,13 @@ class TradingAlgorithm(object):
then set self.mavg to some value and it will be recorded.
"""
if isinstance(names, basestring):
if not isinstance(names, list):
names = [names]
for name in names:
if not isinstance(name, basestring):
raise TypeError("record_variables expects only strings")
if self.initialized:
raise Exception(MESSAGES.ERRORS.CALL_RECORD_VARIABLES_POST_INIT)