def main(): all_stocks = stocks.StockCollection() LoadCSV("march1.csv", all_stocks) LoadCSV("march2.csv", all_stocks) LoadCSV("march3.csv", all_stocks) LoadCSV("march4.csv", all_stocks) LoadCSV("march5.csv", all_stocks) LoadTriplet("feb1.trp", all_stocks) LoadTriplet("feb2.trp", all_stocks) LoadTriplet("feb3.trp", all_stocks) LoadTriplet("feb4.trp", all_stocks) stock = all_stocks.get_stock("ADV") volume = stocks.AverageVolume() stock.analyse(volume) print("Average Volume of ADV is", volume.result()) high_low = HighLow() stock.analyse(high_low) print("Highest & Lowest trading price of ADV is", high_low.result()) moving_average = MovingAverage(10) stock.analyse(moving_average) print("Moving average of ADV over last 10 days is {0:.2f}".format( moving_average.result())) gap_up = GapUp(0.011) stock = all_stocks.get_stock("YOW") stock.analyse(gap_up) print("Last gap up date of YOW is", gap_up.result().get_date())
def setUp(self): """ Setup work before each test """ self.all_stocks = stocks.StockCollection() sa.LoadCSV(TEST_FILES["march1.csv"], self.all_stocks) sa.LoadCSV(TEST_FILES["march2.csv"], self.all_stocks) sa.LoadCSV(TEST_FILES["march3.csv"], self.all_stocks) sa.LoadCSV(TEST_FILES["march4.csv"], self.all_stocks) sa.LoadCSV(TEST_FILES["march5.csv"], self.all_stocks) sa.LoadTriplet(TEST_FILES["feb1.trp"], self.all_stocks) sa.LoadTriplet(TEST_FILES["feb2.trp"], self.all_stocks) sa.LoadTriplet(TEST_FILES["feb3.trp"], self.all_stocks) sa.LoadTriplet(TEST_FILES["feb4.trp"], self.all_stocks)
def example_usage(): # Set up stock market all_stocks = stocks.StockCollection() #loading in data LoadCSV("data_files/march1.csv", all_stocks) LoadCSV("data_files/march2.csv", all_stocks) LoadCSV("data_files/march3.csv", all_stocks) LoadCSV("data_files/march4.csv", all_stocks) LoadCSV("data_files/march5.csv", all_stocks) LoadTriplet("data_files/feb1.trp", all_stocks) LoadTriplet("data_files/feb2.trp", all_stocks) LoadTriplet("data_files/feb3.trp", all_stocks) LoadTriplet("data_files/feb4.trp", all_stocks) #example of a class that analyzers stock volume = stocks.AverageVolume() stock = all_stocks.get_stock("ADV") #All stuff below is gonna analyse ADV stock.analyse(volume) print("Average Volume of ADV is", volume.result()) #testing HighLow Class high_low = HighLow() stock.analyse(high_low) print("Highest & Lowest trading price of ADV is", high_low.result()) #testing MovingAverage Class moving_average = MovingAverage(10) stock.analyse(moving_average) print("Moving average of ADV over last 10 days is {0:.2f}".format( moving_average.result())) #testing GapUp class gap_up = GapUp(0.011) stock = all_stocks.get_stock("YOW") stock.analyse(gap_up) print("Last gap up date of YOW is", gap_up.result().get_date())
def setUp(self): """ Setup work before each test """ self.all_stocks = stocks.StockCollection() sa.LoadCSV(TEST_FILES["march1_small.csv"], self.all_stocks)
def setUp(self): """ Setup work before each test """ self.all_stocks = stocks.StockCollection() sa.LoadTriplet(TEST_FILES['feb1_small.trp'], self.all_stocks)
def setUp(self): """ Setup work before each test """ self.all_stocks = stocks.StockCollection()
import stocks some_stocks = stocks.StockCollection() stock = some_stocks.get_stock("1AD") day_data = stocks.TradingData("20170227", 0.225, 0.225, 0.225, 0.225, 19478) stock.add_day_data(day_data) day_data = stocks.TradingData("20170306", 0.22, 0.225, 0.22, 0.225, 24259) stock.add_day_data(day_data) stock = some_stocks.get_stock("ABU") day_data = stocks.TradingData("20170227", 0.11, 0.115, 0.11, 0.115, 262306) stock.add_day_data(day_data) day_data = stocks.TradingData("20170306", 0.115, 0.115, 0.115, 0.115, 86782) stock.add_day_data(day_data) day_data = stocks.TradingData("20170313", 0.105, 0.105, 0.105, 0.105, 75025) stock.add_day_data(day_data) stock = some_stocks.get_stock("AGL") day_data = stocks.TradingData("20170227", 23.6, 23.92, 23.6, 23.85, 1433574) stock.add_day_data(day_data) day_data = stocks.TradingData("20170306", 24.7, 24.88, 24.4, 24.86, 1156183) stock.add_day_data(day_data) day_data = stocks.TradingData("20170313", 25.48, 25.53, 25.32, 25.45, 1892685) stock.add_day_data(day_data)
end = open + 6 temp = [] code = '-1' for line in data[open:end]: line = line.strip('\r\n') items = line.split(':') code = items[0] temp.append(items[2]) tradingData = stocks.TradingData(temp[0], float(temp[1]), float(temp[2]), float(temp[3]), float(temp[4]), int(temp[5])) stock = stocks.Stock(code) stock.add_day_data(tradingData) self._stocks.set_stock(code, stock) if __name__ == "__main__": all_stocks = stocks.StockCollection() """LoadTriplet("data_files/feb1.trp", all_stocks) LoadTriplet("data_files/feb2.trp", all_stocks) LoadTriplet("data_files/feb3.trp", all_stocks) LoadTriplet("data_files/feb4.trp", all_stocks)""" LoadCSV("data_files/march1.csv", all_stocks) LoadCSV("data_files/march2.csv", all_stocks) LoadCSV("data_files/march3.csv", all_stocks) LoadCSV("data_files/march4.csv", all_stocks) LoadCSV("data_files/march5.csv", all_stocks) volume = stocks.AverageVolume() stock = all_stocks.get_stock("ADV") stock.analyse(volume) print("Average Volume of ADV is", volume.result())