Python Portfolio Examples

Example #1

File: test_port5_pytest.py Project: br3ndonland/test3

def test_sell():
    p = Portfolio()
    p.buy("MSFT", 100, 27.0)
    p.buy("DELL", 100, 17.0)
    p.buy("ORCL", 100, 34.0)
    p.sell("MSFT", 50)
    assert p.cost() == 6450

Example #2

 def test_dont_own_it(self):
     p = Portfolio()  # What, again!?!?
     p.buy("MSFT", 100, 27.0)
     p.buy("DELL", 100, 17.0)
     p.buy("ORCL", 100, 34.0)
     with self.assertRaises(ValueError):
         p.sell("IBM", 1)

Example #3

 def test_not_enough(self):
     p = Portfolio()  # Didn't I just do this?
     p.buy("MSFT", 100, 27.0)
     p.buy("DELL", 100, 17.0)
     p.buy("ORCL", 100, 34.0)
     with self.assertRaises(ValueError):
         p.sell("MSFT", 200)

Example #4

 def test_sell(self):
     p = Portfolio()
     p.buy("MSFT", 100, 27.0)
     p.buy("DELL", 100, 17.0)
     p.buy("ORCL", 100, 34.0)
     p.sell("MSFT", 50)
     self.assertCostEqual(p, 6450)

Example #5

File: test_port5_pytest.py Project: br3ndonland/test3

def test_dont_own_it():
    p = Portfolio()  # What, again!?!?
    p.buy("MSFT", 100, 27.0)  # |
    p.buy("DELL", 100, 17.0)  # |
    p.buy("ORCL", 100, 34.0)  # /
    with pytest.raises(ValueError):
        p.sell("IBM", 1)

Example #6

File: test_port5_pytest.py Project: br3ndonland/test3

def test_not_enough():
    p = Portfolio()  # Didn't I just do this?
    p.buy("MSFT", 100, 27.0)  # |
    p.buy("DELL", 100, 17.0)  # |
    p.buy("ORCL", 100, 34.0)  # /
    with pytest.raises(ValueError):
        p.sell("MSFT", 200)

Example #7

File: test_port6_unittest.py Project: br3ndonland/test3

class PortfolioSellTest(PortfolioTestCase):
    # Invoked before each test method
    def setUp(self):
        self.p = Portfolio()
        self.p.buy("MSFT", 100, 27.0)
        self.p.buy("DELL", 100, 17.0)
        self.p.buy("ORCL", 100, 34.0)

    def test_sell(self):
        self.p.sell("MSFT", 50)
        self.assertCostEqual(self.p, 6450)

    def test_not_enough(self):
        with self.assertRaises(ValueError):
            self.p.sell("MSFT", 200)

    def test_dont_own_it(self):
        with self.assertRaises(ValueError):
            self.p.sell("IBM", 1)

Example #8

File: test_port6_pytest.py Project: br3ndonland/test3

def test_empty():
    p = Portfolio()
    assert p.cost() == 0.0

Example #9

File: test_port6_pytest.py Project: br3ndonland/test3

def test_bad_input():
    p = Portfolio()
    with pytest.raises(TypeError):
        p.buy("IBM")

Example #10

File: test_port6_pytest.py Project: br3ndonland/test3

def test_buy_two_stocks():
    p = Portfolio()
    p.buy("IBM", 100, 176.48)
    p.buy("HPQ", 100, 36.15)
    assert p.cost() == 21263.0

Example #11

File: test_port6_pytest.py Project: br3ndonland/test3

def test_buy_one_stock():
    p = Portfolio()
    p.buy("IBM", 100, 176.48)
    assert p.cost() == 17648.0

Example #12

 def test_bad_input(self):
     p = Portfolio()
     with self.assertRaises(TypeError):
         p.buy("IBM")

Example #13

 def test_buy_two_stocks(self):
     p = Portfolio()
     p.buy("IBM", 100, 176.48)
     p.buy("HPQ", 100, 36.15)
     self.assertCostEqual(p, 21263.0)

Example #14

 def test_buy_one_stock(self):
     p = Portfolio()
     p.buy("IBM", 100, 176.48)
     self.assertCostEqual(p, 17648.0)

Example #15

 def test_empty(self):
     p = Portfolio()
     self.assertCostEqual(p, 0.0)

Example #16

File: test_port6_unittest.py Project: br3ndonland/test3

 def setUp(self):
     self.p = Portfolio()
     self.p.buy("MSFT", 100, 27.0)
     self.p.buy("DELL", 100, 17.0)
     self.p.buy("ORCL", 100, 34.0)