From f914c53790790af8cab3f126556a233fe2d6e05b Mon Sep 17 00:00:00 2001 From: Richard Frank Date: Mon, 11 Feb 2013 18:07:05 -0500 Subject: [PATCH] Added parameter type validation to record_variables --- zipline/algorithm.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/zipline/algorithm.py b/zipline/algorithm.py index 625459df..839812ff 100644 --- a/zipline/algorithm.py +++ b/zipline/algorithm.py @@ -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)