def test_add(self): tmp = self.frac1+self.frac1 print (type(tmp)) self.assertEqual(Frac.__cmp__((self.frac1 + self.frac1), Frac(4, 3)),True) self.assertEqual(Frac.__cmp__(self.frac1 + (-self.frac2), Frac(-2, 3)),True) self.assertEqual(Frac.__cmp__(self.frac4 + 1, Frac(3, 3)), True) self.assertEqual(Frac.__cmp__(1 + self.frac5, Frac(3, 5)), True)
class TestFracs(unittest.TestCase):
def setUp(self):
self.one = Frac(1, 2)
self.two = Frac(1, 4)
self.three = Frac(-1, 6)
self.four = Frac(0, 6)
self.five = Frac(-2, 5)
def test_cmp(self):
self.assertEqual(self.three.__cmp__(self.four), -1)
self.assertEqual(self.three.__cmp__(self.three), 0)
self.assertEqual(self.three.__cmp__(self.five), 1)
def test_add(self):
self.assertEqual(self.one + self.two, Frac(3, 4))
self.assertEqual(self.one + self.four, Frac(3, 6))
self.assertEqual(self.one + self.five, Frac(1, 10))
self.assertEqual(self.one + 2, Frac(5, 2))
self.assertEqual(2 + self.one, Frac(5, 2))
def test_sub(self):
self.assertEqual(self.one - self.two, Frac(1, 4))
self.assertEqual(2 - self.one, Frac(3, 2))
self.assertEqual(self.one - 1, Frac(-1, 2))
def test_mul(self):
self.assertEqual(self.two * self.two, Frac(1, 16))
self.assertEqual(self.one * 5, Frac(5, 2))
self.assertEqual(5 * self.one, Frac(5, 2))
def test_div(self):
self.assertEqual(self.three / self.two, Frac(-2, 3))
self.assertEqual(self.one / 3, Frac(1, 6))
# self.assertEqual(self.one/0, ???)
self.assertEqual(3 / self.one, Frac(6, 1))
#self.assertRaises(ValueError, Frac, self.one/0)
def test_pos(self):
self.assertEqual(self.one.__pos__(), Frac(1, 2))
def test_neg(self):
self.assertEqual(self.one.__neg__(), Frac(-1, 2))
def test_invert(self):
self.assertEqual(self.two.__invert__(), Frac(4, 1))
def test_is_positive(self):
self.assertEqual(self.one.is_positive(), True)
self.assertEqual(self.five.is_positive(), False)
def test_is_zero(self):
self.assertEqual(self.two.is_zero(), False)
self.assertEqual(self.four.is_zero(), True)
def test_frac_to_float(self):
self.assertEqual(self.five.frac2float(), -0.4)
self.assertEqual(self.two.frac2float(), 0.25)
def test_invert(self): self.assertEqual(Frac.__cmp__(~self.frac1, Frac(3, 2)), True) self.assertEqual(Frac.__cmp__(-self.frac5, Frac(2, 5)), True)
def test_neg(self): self.assertEqual(Frac.__cmp__(-self.frac1, Frac(-2, 3)), True) self.assertEqual(Frac.__cmp__(-self.frac2, Frac(-4, 3)), True) self.assertEqual(Frac.__cmp__(-self.frac3, Frac(-2, 1)), True)
def test_div(self): self.assertEqual(Frac.__cmp__(self.frac1/self.frac1, Frac(6, 6)), True) self.assertEqual(Frac.__cmp__(self.frac5/3, Frac(-2, 15)), True) self.assertEqual(Frac.__cmp__(2/self.frac5, Frac(10, -2)), True) with self.assertRaises(ValueError): self.frac2 / 0
def test_mul(self): self.assertEqual(Frac.__cmp__(self.frac1*self.frac2, Frac(8, 9)), True) self.assertEqual(Frac.__cmp__(self.frac3*self.frac5, Frac(-4, 5)), True) self.assertEqual(Frac.__cmp__(self.frac1*2, Frac(4, 3)), True) self.assertEqual(Frac.__cmp__(2*self.frac1, Frac(4, 3)), True)
def test_sub(self): self.assertEqual(Frac.__cmp__(self.frac1-self.frac2, Frac(-2, 3)),True) self.assertEqual(Frac.__cmp__(self.frac2 -self.frac1, Frac(2, 3)), True) self.assertEqual(Frac.__cmp__(self.frac4 - 1, Frac(-3, 3)), True) self.assertEqual(Frac.__cmp__(1 - self.frac5, Frac(7, 5)), True)
def test_cmp(self): self.assertEqual(Frac.__cmp__(self.frac1, self.frac1), True) self.assertEqual(Frac.__cmp__(self.frac1, self.frac2), False)
