29 lines
721 B
Python
29 lines
721 B
Python
# test_stock.py
|
|
|
|
import unittest
|
|
from . import stock
|
|
|
|
class TestStock(unittest.TestCase):
|
|
def test_create(self):
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
|
self.assertEqual(s.name, 'GOOG')
|
|
self.assertEqual(s.shares, 100)
|
|
self.assertEqual(s.price, 490.1)
|
|
|
|
def test_cost(self):
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
|
self.assertEqual(s.cost, 49010.0)
|
|
|
|
def test_sell(self):
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
|
s.sell(25)
|
|
self.assertEqual(s.shares, 75)
|
|
|
|
def test_shares_check(self):
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
|
with self.assertRaises(TypeError):
|
|
s.shares = '100'
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|