Add test script

Author: Brian

requirements.txt

@@ -1 +1,2 @@
 requests
+colorama

test.py

@@ -0,0 +1,107 @@
+from colorama import Fore, Style
+from tradingview_ta import TA_Handler, Interval
+import tradingview_ta, requests
+
+print("------------------------------------------------")
+print("Testing {}Tradingview-TA{} v{}{}".format(Fore.CYAN, Fore.MAGENTA, tradingview_ta.__version__, Style.RESET_ALL))
+print("This test is {}semi-automatic{}. Please compare with tradingview's data manually.".format(Fore.LIGHTRED_EX, Style.RESET_ALL))
+print("------------------------------------------------")
+
+COUNT = 5
+success = 0
+
+print("{}#0{} {}Testing invalid symbol{}".format(Fore.BLUE, Style.RESET_ALL, Fore.LIGHTBLUE_EX, Style.RESET_ALL))
+handler = TA_Handler(
+    symbol="ThisSymbolIsInvalid",
+    interval="1m",
+    screener="america",
+    exchange="NASDAQ"
+)
+try:
+    analysis = handler.get_analysis()
+    if analysis:
+        print("{}#0{} Invalid symbol test {}failed{}. No exception occured.".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL))
+except Exception as e:
+    if str(e) == "Exchange or symbol not found.":
+        print("{}#0{} Invalid symbol test {}success{}.".format(Fore.BLUE, Style.RESET_ALL, Fore.GREEN, Style.RESET_ALL))
+        success += 1
+    else:
+        print("{}#0{} Invalid symbol test {}failed{}. An exception occured, but the symbol is valid.".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL))
+
+
+print("{}#1{} {}Testing invalid exchange{}".format(Fore.BLUE, Style.RESET_ALL, Fore.LIGHTBLUE_EX, Style.RESET_ALL))
+handler = TA_Handler(
+    symbol="TSLA",
+    interval="1m",
+    screener="america",
+    exchange="binance"
+)
+try:
+    analysis = handler.get_analysis()
+    if analysis:
+        print("{}#1{} Invalid exchange test {}failed{}. No exception occured.".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL))
+except Exception as e:
+    if str(e) == "Exchange or symbol not found.":
+        print("{}#1{} Invalid exchange test {}success{}.".format(Fore.BLUE, Style.RESET_ALL, Fore.GREEN, Style.RESET_ALL))
+        success += 1
+    else:
+        print("{}#1{} Invalid exchange test {}failed{}. An exception occured, but symbol is valid.".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL))
+
+print("{}#2{} {}Testing timeout{}".format(Fore.BLUE, Style.RESET_ALL, Fore.LIGHTBLUE_EX, Style.RESET_ALL))
+handler = TA_Handler(
+    symbol="AAPL",
+    interval=Interval.INTERVAL_1_DAY,
+    screener="america",
+    exchange="NASDAQ",
+    timeout=0.0001
+)
+try:
+    analysis = handler.get_analysis()
+    if analysis:
+        print("{}#2{} Timeout test {}failed{}.".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL))
+except Exception as e:
+    if type(e) == requests.exceptions.ConnectTimeout:
+        print("{}#2{} Timeout test {}success{}.".format(Fore.BLUE, Style.RESET_ALL, Fore.GREEN, Style.RESET_ALL))
+        success += 1
+
+
+print("{}#3{} {}Testing invalid interval{}".format(Fore.BLUE, Style.RESET_ALL, Fore.LIGHTBLUE_EX, Style.RESET_ALL))
+handler = TA_Handler(
+    symbol="TSLA",
+    interval="1 minute",
+    screener="america",
+    exchange="NASDAQ"
+)
+try:
+    analysis = handler.get_analysis()
+    if analysis and input('{}#3{} Did you see a "defaulting to 1 day" {}warning{}? (Y/N) '.format(Fore.BLUE, Style.RESET_ALL, Fore.YELLOW, Style.RESET_ALL)).lower() == "y":
+        print("{}#3{} Invalid interval test {}success{}.".format(Fore.BLUE, Style.RESET_ALL, Fore.GREEN, Style.RESET_ALL))
+        success += 1
+    else:
+        print("{}#3{} Invalid interval test {}failed{}".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL))
+
+except Exception as e:
+    print("{}#3{} Invalid interval test {}failed{}. An exception occured: {}".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL, e))
+
+print("{}#4{} {}Testing stock (NASDAQ:AAPL){}".format(Fore.BLUE, Style.RESET_ALL, Fore.LIGHTBLUE_EX, Style.RESET_ALL))
+handler = TA_Handler(
+    symbol="AAPL",
+    interval=Interval.INTERVAL_1_DAY,
+    screener="america",
+    exchange="NASDAQ"
+)
+try:
+    analysis = handler.get_analysis()
+    print("{}#4{} Please compare with {}https://www.tradingview.com/symbols/NASDAQ-AAPL/technicals/{}.".format(Fore.BLUE, Style.RESET_ALL, Fore.LIGHTMAGENTA_EX, Style.RESET_ALL))
+    print("{}#4{} (Summary) Rec: {}, Sell: {}, Neutral: {}, Buy: {}".format(Fore.BLUE, Style.RESET_ALL, analysis.summary["RECOMMENDATION"], analysis.summary["SELL"], analysis.summary["NEUTRAL"], analysis.summary["BUY"]))
+    if input("{}#4{} Are the results the same? (Y/N) ".format(Fore.BLUE, Style.RESET_ALL)).lower() == "y":
+        print("{}#4{} Stock test {}success{}.".format(Fore.BLUE, Style.RESET_ALL, Fore.GREEN, Style.RESET_ALL))
+        success += 1
+    else:
+        print("{}#4{} Stock test {}failed{}".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL))
+except Exception as e:
+    print("{}#4{} Stock test {}failed{}. An exception occured: {}".format(Fore.BLUE, Style.RESET_ALL, Fore.RED, Style.RESET_ALL, e))
+
+
+print("------------------------------------------------")
+print("Test finished. Result: {}{}/{}{}.".format(Fore.LIGHTWHITE_EX ,success, COUNT, Style.RESET_ALL))