clean up

areed1192 · Jun 18, 2020 · 67b9ff2 · 67b9ff2
1 parent aa0a457
commit 67b9ff2
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 10 deletions.
diff --git a/samples/api_streaming.py b/samples/api_streaming.py
@@ -23,6 +23,9 @@
     append_mode=True
 )
 
+# Account Activity
+TDStreamingClient.account_activity()
+
 # Actives
 TDStreamingClient.actives(service='ACTIVES_NASDAQ', venue='NASDAQ', duration='ALL')
 
@@ -59,9 +62,6 @@
 # News Headline
 TDStreamingClient.news_headline(symbols=['AAPL', 'SPY'], fields=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
 
-# Account Activity
-TDStreamingClient.account_activity()
-
 # Level Two Quotes
 TDStreamingClient.level_two_quotes(symbols = ['IBM'], fields = [0,1,2])
 

diff --git a/td/defaults.py b/td/defaults.py
@@ -9,13 +9,6 @@
 from typing import Union
 from typing import Optional
 
-# if sys.version_info >= (3, 5):
-#     home_dir = str(pathlib.Path.home())
-# else:
-#     home_dir = os.path.expanduser('~')
-
-# default_dir = os.path.join(home_dir, '.tdunofficial')
-
 class StatePath(type(pathlib.Path())):
 
     def __init__(self, credentials_file: str = None):

diff --git a/tests/unit/test_stream.py b/tests/unit/test_stream.py
@@ -43,6 +43,12 @@ def test_create_stream_session(self):
 
         self.assertIsInstance(self.stream_session, TDStreamerClient)
 
+    def test_create_account_activity(self):
+        """Test subscribing to account activity."""
+
+        self.stream_session.account_activity()
+        self.assertIn('ACCT_ACTIVITY', self.stream_session.data_requests['requests'][0]['service'])
+
     def test_subscribe_level_one_quotes(self):
         """Test subscribing to Level One Quotes."""