Files
OpenBB/tests/test_pred_neural_networks.py
2021-03-12 09:19:33 -08:00

27 lines
825 B
Python

""" fundamental_analysis/yahoo_finance_api.py tests """
import unittest
try:
from gamestonk_terminal.prediction_techniques.neural_networks import mlp
except ModuleNotFoundError as e:
print("One of the optional packages seems to be missing")
print(e)
print("Skipping the test")
from gamestonk_terminal import feature_flags as gtff
class TestPredNeuralNetworks(unittest.TestCase):
def test_mlp(self):
if not gtff.ENABLE_PREDICT:
return
# Fix: need to come up with a better way of doing this
try:
# Fix: need to move loading of ticker data into a df somewhere
mlp([], "TLSA", None)
except NameError as e:
print("One of the optional packages seems to be missing")
print(e)
print("Skipping the test")