Add test for realtime. #9

toomore · Jan 10, 2014 · bb36eea · bb36eea
1 parent eccd55d
commit bb36eea
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test_unittest.py b/test_unittest.py
@@ -74,6 +74,12 @@ def test_realtime():
         assert real_time.real['no'] == '2618'
         real_time = grs.RealtimeWeight()
         assert real_time.real['no'] == '1'
+        real_time = grs.RealtimeStock('0050')
+        assert real_time.real['no'] == '0050'
+        try:
+            real_time = grs.RealtimeStock(0050)
+        except AssertionError:
+            pass
 
     @staticmethod
     def test_countdown():