73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
import unittest
|
|
|
|
import stock
|
|
|
|
|
|
class TestStock(unittest.TestCase):
|
|
def test_create(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
self.assertEquals(s.name, "GOOG")
|
|
self.assertEquals(s.shares, 100)
|
|
self.assertEquals(s.price, 490.1)
|
|
|
|
def test_create_keyword(self):
|
|
s = stock.Stock(name="GOOG", shares=100, price=490.1)
|
|
self.assertEquals(s.name, "GOOG")
|
|
self.assertEquals(s.shares, 100)
|
|
self.assertEquals(s.price, 490.1)
|
|
|
|
def test_from_row(self):
|
|
s = stock.Stock.from_row(("GOOG", 100, 490.1))
|
|
self.assertEquals(s.name, "GOOG")
|
|
self.assertEquals(s.shares, 100)
|
|
self.assertEquals(s.price, 490.1)
|
|
|
|
def test_cost(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
self.assertEquals(s.cost, 49010)
|
|
|
|
def test_sell(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
s.sell(50)
|
|
self.assertEqual(s.shares, 50)
|
|
|
|
def test_repr(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
self.assertEquals(repr(s), "Stock('GOOG', 100, 490.1)")
|
|
|
|
def test_eq(self):
|
|
s1 = stock.Stock("GOOG", 100, 490.1)
|
|
s2 = stock.Stock("GOOG", 100, 490.1)
|
|
s3 = stock.Stock("GOOG", 50, 490.1)
|
|
self.assertEqual(s1, s2)
|
|
self.assertNotEqual(s1, s3)
|
|
|
|
def test_shares_type(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
with self.assertRaises(TypeError):
|
|
s.shares = "50"
|
|
|
|
def test_shares_value(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
with self.assertRaises(ValueError):
|
|
s.shares = -50
|
|
|
|
def test_price_type(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
with self.assertRaises(TypeError):
|
|
s.price = "50"
|
|
|
|
def test_price_value(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
with self.assertRaises(ValueError):
|
|
s.price = -90.9
|
|
|
|
def test_bad_attr(self):
|
|
s = stock.Stock("GOOG", 100, 490.1)
|
|
with self.assertRaises(AttributeError):
|
|
s.share = "foo"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|