update tests

rochi88 · Dec 31, 2024 · 3779f3c · 3779f3c
1 parent 5e30280
commit 3779f3c
Show file tree

Hide file tree

Showing 9 changed files with 51 additions and 70 deletions.
diff --git a/tests/test_company_ins.py b/tests/test_company_ins.py
diff --git a/tests/test_dsex.py b/tests/test_dsex.py
diff --git a/tests/test_latest_pe.py b/tests/test_latest_pe.py
diff --git a/tests/test_market.py b/tests/test_market.py
@@ -0,0 +1,34 @@
+# _*_ coding:utf-8 _*_
+'''
+Created on 2024-Dec-31
+@author: Raisul Islam
+'''
+import unittest
+import datetime as dt
+from bdshare import get_latest_pe, get_market_inf, get_market_depth_data, get_market_inf_more_data
+
+
+class Test(unittest.TestCase):
+
+    def test_get_latest_pe(self):
+        df = get_latest_pe()
+        print(df.to_string())
+
+    def test_get_market_inf(self):
+        df = get_market_inf()
+        print(df.to_string())
+
+    def test_get_market_depth(self):
+        df = get_market_depth_data('AAMRANET')
+        print(df.to_string())
+
+    def test_get_market_inf_more_data(self):
+        start = dt.datetime.now().date() - dt.timedelta(days=2 * 365)
+        end = dt.datetime.now().date()
+        df = get_market_inf_more_data(start, end, index='date')
+        print(df.to_string())
+        print(df.dtypes)
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/tests/test_mkt_depth.py b/tests/test_mkt_depth.py
diff --git a/tests/test_mkt_inf.py b/tests/test_mkt_inf.py
diff --git a/tests/test_news.py b/tests/test_news.py
@@ -4,7 +4,6 @@
 @author: Raisul Islam
 '''
 import unittest
-# import datetime as dt
 from bdshare import get_agm_news, get_all_news
 
 
@@ -15,7 +14,6 @@ def test_get_agm_news(self):
         print(df.to_string())
 
     def test_get_all_news(self):
-        # end = dt.datetime.now().date()
         df = get_all_news('BATBC')
         print(df.to_string())
 

diff --git a/tests/test_random.py b/tests/test_random.py
@@ -1,10 +1,17 @@
 # _*_ coding:utf-8 _*_
 '''
-Created on 2021-May-29
+Created on 2024-Dec-31
 @author: Raisul Islam
 '''
-from bdshare import get_current_trade_data
 
-df = get_current_trade_data()
-print(df.to_json(orient='records'))
-print(df.to_string())
+from bdshare import get_company_inf
+
+df = get_company_inf('GP')
+
+print(f'Total tables: {len(df)}')
+
+# for i in range(len(df)):
+#     print(f'Table {i}')
+#     print(df[i].to_string())
+
+print(df[8])
diff --git a/tests/test_trading.py b/tests/test_trading.py
@@ -5,11 +5,15 @@
 '''
 import unittest
 import datetime as dt
-from bdshare import get_current_trade_data, get_basic_hist_data, get_market_inf_more_data
+from bdshare import get_dsex_data, get_current_trade_data, get_basic_hist_data, get_market_inf_more_data
 
 
 class Test(unittest.TestCase):
 
+    def test_get_dsex_data(self):
+        df = get_dsex_data()
+        print(df.to_string())
+
     def test_get_current_trade_data(self):
         df = get_current_trade_data()
         print(df.to_string())