def test_date_before_start_keyerror_should_appear(self): start = "2010-01-01" date = '2009-05-28' tickers = list(gt.get_biggest_n_tickers(40)) b = Analyzer(tickers, start) #self.assertRaises(KeyError, lambda: b.winners(date, 25, 5)) self.assertEqual(b.winners(date, 25, 5), [])
def test_ticker_not_momentum_before_list(self): start_date = "2010-01-01" ranking_period = 25 n = 5 listeddate = "2010-06-28" s = 'TSLA' tickers = list(gt.get_biggest_n_tickers(40)) b = Analyzer(tickers, start_date) self.assertEqual( b.appeartimes(start_date, ranking_period, n, listeddate, s, volume_filter=False), (0, 0))
from Analyzer import Analyzer from get_all_tickers import get_tickers as gt if __name__ == '__main__': tickers = gt.get_biggest_n_tickers(40) analyzer = Analyzer(tickers, start='2010-01-01') print(tickers)
def test_date_before_oneperiod(self): start = "2010-01-01" date = '2010-01-28' tickers = list(gt.get_biggest_n_tickers(40)) b = Analyzer(tickers, start) self.assertEqual(b.winners(date, 25, 5), [])
from django import forms from get_all_tickers.get_tickers import get_biggest_n_tickers #set the default values for the index page tweet_counts = [(0, 'Tweet Counts'), (10, '10 tweets per hour'), (25, '25 tweets per hour'), (50, '50 tweets per hour'), (75, '75 tweets per hour'), (100, '100 tweets per hour')] tweet_days = [(0, 'Tweet Days'), (1, 'last 1 day'), (2, 'last 2 day'), (3, 'last 3 day'), (4, 'last 4 day'), (5, 'last 5 day'), (6, 'last 6 day'), (7, 'last week')] #setting value of first n tickers and extracting stock tickers n = 100 n_tickers = sorted(get_biggest_n_tickers(n, sectors=None)) # use comprehensive list to create the list of tuples tickers_choice = [(ticker, ticker) for ticker in n_tickers] tickers_choice.insert(0, ('0', 'Select Stocks...')) #django form class StockSentimentAnalystForm(forms.Form): tickers = forms.ChoiceField(choices=tickers_choice, label='') tweet_count = forms.ChoiceField(label='', choices=tweet_counts, help_text='Tweet Counts') tweet_day = forms.ChoiceField(label='', choices=tweet_days, help_text='Tweet Days') tickers.widget.attrs.update({'class': 'custom-select'}) tweet_count.widget.attrs.update({'class': 'custom-select'})
def build_top_n_tickers(self): # Get a list of top N US stock tickers print(f"Start fetching top {self.num_company} stock tickers") self.list_top_n_tickers = get_tickers.get_biggest_n_tickers(self.num_company) print(f"Finished")