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())
