130 lines
3.4 KiB
Markdown
130 lines
3.4 KiB
Markdown
|
# Exercise 5.6 - Solution
|
||
|
|
|
||
|
|
## (b) Unit testing
|
||
|
|
|
||
|
|
```python
|
||
|
|
# teststock.py
|
||
|
|
|
||
|
|
import stock
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
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_create_keyword(self):
|
||
|
|
s = stock.Stock(name='GOOG', shares=100, price=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_from_row(self):
|
||
|
|
s = stock.Stock.from_row(['GOOG','100','490.1'])
|
||
|
|
self.assertEqual(s.name, 'GOOG')
|
||
|
|
self.assertEqual(s.shares, 100)
|
||
|
|
self.assertEqual(s.price, 490.1)
|
||
|
|
|
||
|
|
def test_repr(self):
|
||
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
self.assertEqual(repr(s), "Stock('GOOG', 100, 490.1)")
|
||
|
|
|
||
|
|
def test_eq(self):
|
||
|
|
a = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
b = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
self.assertTrue(a==b)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|
||
|
|
```
|
||
|
|
|
||
|
|
## (c) Testing for Errors
|
||
|
|
|
||
|
|
```python
|
||
|
|
# teststock.py
|
||
|
|
|
||
|
|
import stock
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
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_create_keyword(self):
|
||
|
|
s = stock.Stock(name='GOOG', shares=100, price=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_from_row(self):
|
||
|
|
s = stock.Stock.from_row(['GOOG','100','490.1'])
|
||
|
|
self.assertEqual(s.name, 'GOOG')
|
||
|
|
self.assertEqual(s.shares, 100)
|
||
|
|
self.assertEqual(s.price, 490.1)
|
||
|
|
|
||
|
|
def test_repr(self):
|
||
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
self.assertEqual(repr(s), "Stock('GOOG', 100, 490.1)")
|
||
|
|
|
||
|
|
def test_eq(self):
|
||
|
|
a = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
b = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
self.assertTrue(a==b)
|
||
|
|
|
||
|
|
# Tests for failure conditions
|
||
|
|
def test_shares_badtype(self):
|
||
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
with self.assertRaises(TypeError):
|
||
|
|
s.shares = '50'
|
||
|
|
|
||
|
|
def test_shares_badvalue(self):
|
||
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
with self.assertRaises(ValueError):
|
||
|
|
s.shares = -50
|
||
|
|
|
||
|
|
def test_price_badtype(self):
|
||
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
with self.assertRaises(TypeError):
|
||
|
|
s.price = '45.23'
|
||
|
|
|
||
|
|
def test_price_badvalue(self):
|
||
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
with self.assertRaises(ValueError):
|
||
|
|
s.price = -45.23
|
||
|
|
|
||
|
|
def test_bad_attribute(self):
|
||
|
|
s = stock.Stock('GOOG', 100, 490.1)
|
||
|
|
with self.assertRaises(AttributeError):
|
||
|
|
s.share = 100
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|
||
|
|
```
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
[Back](ex5_6.md)
|