from Portfolio import Portfolio, Stock, MutualFund p = Portfolio() p.addCash(250) # Always works p.withdrawCash(50) # Should work p.withdrawCash(3000) # Insufficient balance print(p) p.history() s1 = Stock(20, "AAPL") s2 = Stock(35, "GOOG") p.buyStock(3, s1) # Should work p.buyStock(2, s2) # Should work p.buyStock(100, s1) # Insufficient balance print(p) p.history() p.sellStock("AAPL", 2) # Should work p.sellStock("GOOG", 10) # Not enough GOOG stock p.sellStock("NASA", 15) # No NASA stock p.addCash(200) mf1 = MutualFund("SNCF") mf2 = MutualFund("DBHN") p.buyMutualFund(36.8, mf1) # Should work p.buyMutualFund(29.45, mf2) # Should work p.buyMutualFund(9000, mf1) # Insufficient balance
##main module from Portfolio import Portfolio from Stock import Stock from MutualFund import MutualFund portfolio = Portfolio() portfolio.addCash(300.50) s = Stock(20, "HFH") portfolio.buyStock(5, s) mf1 = MutualFund("BRT") mf2 = MutualFund("GHT") portfolio.buyMutualFund(10.3, mf1) portfolio.buyMutualFund(2, mf2) print(portfolio) portfolio.sellMutualFund("BRT", 3) portfolio.sellStock("HFH", 1) portfolio.withdrawCash(50) portfolio.history()
from Portfolio import Portfolio from stock import Stock from mutualFund import MutualFund portfolio = Portfolio() #Creates a new portfolio s = Stock(20, "HFH") #Create Stock with price 20 and symbol "HFH" portfolio.addCash(300.50) #Adds cash to the portfolio s = Stock(20, "HFH") #Create Stock with price 20 and symbol "HFH" portfolio.buyStock(5, s) #Buys 5 shares of stock s mf1 = MutualFund("BRT") #Create MF with symbol "BRT" mf2 = MutualFund("GHT") #Create MF with symbol "GHT" portfolio.buyMutualFund(10.3, mf1) #Buys 10.3 shares of "BRT" portfolio.buyMutualFund(2, mf2) #Buys 2 shares of "GHT" print(portfolio) #Prints portfolio #cash: $140.50 #stock: 5 HFH #mutual funds: 10.33 BRT # 2 GHT portfolio.sellMutualFund("BRT", 3) #Sells 3 shares of BRT portfolio.sellStock("HFH", 1) #Sells 1 share of HFH portfolio.withdrawCash(50) #Removes $50 portfolio.history()