def test_average_cost(self):
a = Analyst(1000)
a.buyLong(10.2,2)
a.buyLong(20.66,5)
a.buyLong(30.66, 17)
avg = (30.66*17 + 20.66 * 5 + 10.2 * 2) / (2+5+17)
self.assertAlmostEqual(a.longs.avgCost, avg)
a.sellLong(30, 15)
a.buyLong(40.1, 8)
self.assertAlmostEqual(a.longs.avgCost, 33.09676470588235)
def test_buy_sell_bad(self):
a = Analyst(100)
buy = a.buyLong(50, 3)
self.assertFalse(buy)
self.assertAnalyst(a, 100, 0, 0)
sell = a.sellLong(55, 3)
self.assertFalse(sell)
self.assertAnalyst(a, 100, 0, 0)
def test_buy_sell_long(self):
a = Analyst(150)
a.buyLong(50, 3)
a.sellLong(55, 3)
cash = 150 - (3 * 50) + (3 * 55)
self.assertAnalyst(a, cash, 0, 0)