diff --git a/client.py b/client.py index 40badcd..84b329c 100644 --- a/client.py +++ b/client.py @@ -35,14 +35,16 @@ def getDataPoint(quote): stock = quote['stock'] bid_price = float(quote['top_bid']['price']) ask_price = float(quote['top_ask']['price']) - price = bid_price + price = (bid_price + ask_price)/2 return stock, bid_price, ask_price, price def getRatio(price_a, price_b): """ Get ratio of price_a and price_b """ """ ------------- Update this function ------------- """ """ Also create some unit tests for this function in client_test.py """ - return 1 + if(price_b == 0): + return + return price_a/price_b # Main if __name__ == "__main__": @@ -54,6 +56,7 @@ def getRatio(price_a, price_b): """ ----------- Update to get the ratio --------------- """ for quote in quotes: stock, bid_price, ask_price, price = getDataPoint(quote) + prices[stock] = price print "Quoted %s at (bid:%s, ask:%s, price:%s)" % (stock, bid_price, ask_price, price) - print "Ratio %s" % getRatio(price, price) + print "Ratio %s" % getRatio(prices['ABC'], prices['DEF'])