tests/rate_test.py

   1 #!/usr/bin/env python3
   2
   3 import unittest
   4
   5 from type.rate import Rate
   6 from type.money import Money
   7
   8 import unittest_utils as uu
   9
  10
  11 class TestRate(unittest.TestCase):
  12     def test_basic_utility(self):
  13         my_stock_returns = Rate(percent_change=-20.0)
  14         my_portfolio = 1000.0
  15         self.assertAlmostEqual(
  16             800.0,
  17             my_stock_returns.apply_to(my_portfolio)
  18         )
  19
  20         my_bond_returns = Rate(percentage=104.5)
  21         my_money = Money(500.0)
  22         self.assertAlmostEqual(
  23             Money(522.5),
  24             my_bond_returns.apply_to(my_money)
  25         )
  26
  27         my_multiplier = Rate(multiplier=1.72)
  28         my_nose_length = 3.2
  29         self.assertAlmostEqual(
  30             5.504,
  31             my_multiplier.apply_to(my_nose_length)
  32         )
  33
  34     def test_conversions(self):
  35         x = Rate(104.55)
  36         s = x.__repr__()
  37         y = Rate(s)
  38         self.assertAlmostEqual(x, y)
  39         f = float(x)
  40         z = Rate(f)
  41         self.assertAlmostEqual(x, z)
  42
  43     def test_divide(self):
  44         x = Rate(20.0)
  45         x /= 2
  46         self.assertAlmostEqual(10.0, x)
  47         x = Rate(-20.0)
  48         x /= 2
  49         self.assertAlmostEqual(-10.0, x)
  50
  51     def test_add(self):
  52         x = Rate(5.0)
  53         y = Rate(10.0)
  54         z = x + y
  55         self.assertAlmostEqual(15.0, z)
  56         x = Rate(-5.0)
  57         x += y
  58         self.assertAlmostEqual(5.0, x)
  59
  60     def test_sub(self):
  61         x = Rate(5.0)
  62         y = Rate(10.0)
  63         z = x - y
  64         self.assertAlmostEqual(-5.0, z)
  65         z = y - x
  66         self.assertAlmostEqual(5.0, z)
  67
  68     def test_repr(self):
  69         x = Rate(percent_change=-50.0)
  70         s = x.__repr__(relative=True)
  71         self.assertEqual("-50.000%", s)
  72         s = x.__repr__()
  73         self.assertEqual("+50.000%", s)
  74
  75
  76 if __name__ == '__main__':
  77     unittest.main()