Coverage for python3/tests/test_rh.py: 100%

1# tests/test_rh.py 

2from src.rh import Portfolio 

3from src.rh import Stock 

4from src.rh import CSVParser 

5 

6def test_create_stock(): 

7 portfolio = Portfolio() 

8 portfolio.createStock("AMD") 

9 assert portfolio.positions["AMD"].symbol == "AMD" 

10 assert type(portfolio.positions["AMD"]) == Stock 

11def test_get_stock(): 

12 portfolio = Portfolio() 

13 portfolio.createStock("AMD") 

14 stock = portfolio.getStock("AMD") 

15 assert type(stock) == Stock 

16 assert stock.symbol == "AMD" 

17def test_CSV_obj_creation(): 

18 portfolio = Portfolio() 

19 portfolio.setCSVObj("test.csv", ['GRPN', 'FIT', 'ATVI']) 

20 assert type(portfolio.csvParser) == CSVParser 

21def test_CSV_IGNORELIST_none(): 

22 portfolio = Portfolio() 

23 portfolio.setCSVObj("test.csv") 

24 assert type(portfolio.csvParser) == CSVParser 

25def test_get_portfolio(): 

26 portfolio = Portfolio() 

27 portfolio.setCSVObj("test.csv", ['GRPN', 'FIT', 'ATVI']) 

28 portfolio.getProfile() 

29 assert type(portfolio.csvParser) == CSVParser 

30 assert type(portfolio.getStock("AMD")) == Stock