Add assert for stock_no. #9

toomore · Jan 10, 2014 · 38fcb43 · 38fcb43
1 parent 237a7ad
commit 38fcb43
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/grs/fetch_data.py b/grs/fetch_data.py
@@ -35,6 +35,7 @@ def __init__(self, stock_no, mons=3):
             :stock_no : 股價代碼
             :mons : 擷取近 n 個月的資料
         """
+        assert isinstance(stock_no, str), '`stock_no` must be a string'
         self.__get_mons = 0
         self.__get_no = 0
         self.__info = ()