import unittest
import exceptions

from ledger import amount

class BasicAmountTestCase(unittest.TestCase):
    def testConstructors(self):
        x0 = amount()
        x1 = amount(123456L)
        x2 = amount(123456)
        x3 = amount(123.456)
        x4 = amount(True)
        x5 = amount("123456")
        x6 = amount("123.456")
        x9 = amount(x3)
        x10 = amount(x6)

        self.assertEqual(amount(0L), x0)
        self.assertEqual(x2, x1)
        self.assertEqual(x5, x1)
        self.assertEqual(x6, x3)
        self.assertEqual(x10, x3)
        self.assertEqual(amount(1L), x4)
        self.assertEqual(x10, x9)

    def testNegation(self):
        x0 = amount()
        x1 = amount(-123456L)
        x3 = amount(-123.456)
        x5 = amount("-123456")
        x6 = amount("-123.456")
        x9 = amount(- x3)

        self.assertEqual(amount(0L), x0)
        self.assertEqual(x5, x1)
        self.assertEqual(x6, x3)
        self.assertEqual(- x6, x9)
        self.assertEqual(x3.negated(), x9)

        x10 = amount(x9)
        x10.negate()

        self.assertEqual(x3, x10)

    def testAssignment(self):
        x0 = amount()
        x1  = amount(123456L)
        x2  = amount(123456)
        x3  = amount(123.456)
        x4  = amount(True)
        x5  = amount("123456")
        x6  = amount("123.456")
        x9  = x3
        x10 = amount(x6)

        self.assertEqual(amount(0L), x0)
        self.assertEqual(x2, x1)
        self.assertEqual(x5, x1)
        self.assertEqual(x6, x3)
        self.assertEqual(x10, x3)
        self.assertEqual(amount(1L), x4)
        self.assertEqual(x10, x9)

        x0  = amount()
        x1  = amount(123456L)
        x2  = amount(123456)
        x3  = amount(123.456)
        x4  = amount(True)
        x5  = amount("123456")
        x6  = amount("123.456")
        x9  = x3
        x10 = amount(x6)

        self.assertEqual(amount(0L), x0)
        self.assertEqual(x2, x1)
        self.assertEqual(x5, x1)
        self.assertEqual(x6, x3)
        self.assertEqual(x10, x3)
        self.assertEqual(amount(1L), x4)
        self.assertEqual(x10, x9)

    def testEquality(self):
        x1 = amount(123456L)
        x2 = amount(456789L)
        x3 = amount(333333L)
        x4 = amount(123456.0)
        x5 = amount("123456.0")

        self.assertTrue(x1 == 123456L)
        self.assertTrue(x1 != x2)
        self.assertTrue(x1 == (x2 - x3))
        self.assertTrue(x1 == x4)
        self.assertTrue(x4 == x5)

    def testIntegerAddition(self):
        x1 = amount(123L)
        y1 = amount(456L)

        self.assertEqual(amount(579L), x1 + y1)
        self.assertEqual(amount(579L), x1 + 456L)
        self.assertEqual(amount(579L), 456L + x1)

        x1 += amount(456L)
        self.assertEqual(amount(579L), x1)
        x1 += 456L
        self.assertEqual(amount(1035L), x1)

        x3 = amount(True)
        y3 = amount(True)

        self.assertEqual(amount(2L), x3 + y3)
        self.assertEqual(amount(2L), x3 + True)

        x4 = amount("123456789123456789123456789")

        self.assertEqual(amount("246913578246913578246913578"), x4 + x4)

    def testFractionalAddition(self):
        x1 = amount(123.123)
        y1 = amount(456.456)

        self.assertEqual(amount(579.579), x1 + y1)
        self.assertEqual(amount(579.579), x1 + 456.456)
        self.assertEqual(amount(579.579), 456.456 + x1)

        x1 += amount(456.456)
        self.assertEqual(amount(579.579), x1)
        x1 += 456.456
        self.assertEqual(amount(1036.035), x1)
        x1 += 456L
        self.assertEqual(amount(1492.035), x1)

        x2 = amount("123456789123456789.123456789123456789")

        self.assertEqual(amount("246913578246913578.246913578246913578"), x2 + x2)

    def testIntegerSubtraction(self):
        x1 = amount(123L)
        y1 = amount(456L)

        self.assertEqual(amount(333L), y1 - x1)
        self.assertEqual(amount(-333L), x1 - y1)
        self.assertEqual(amount(23L), x1 - 100L)
        self.assertEqual(amount(-23L), 100L - x1)

        x1 -= amount(456L)
        self.assertEqual(amount(-333L), x1)
        x1 -= 456L
        self.assertEqual(amount(-789L), x1)

        x3 = amount(True)
        y3 = amount(True)

        self.assertEqual(amount(False), x3 - y3)

        x4 = amount("123456789123456789123456789")
        y4 = amount("8238725986235986")

        self.assertEqual(amount("123456789115218063137220803"), x4 - y4)
        self.assertEqual(amount("-123456789115218063137220803"), y4 - x4)

    def testFractionalSubtraction(self):
        x1 = amount(123.123)
        y1 = amount(456.456)

        self.assertEqual(amount(-333.333), x1 - y1)
        self.assertEqual(amount(333.333), y1 - x1)

        x1 -= amount(456.456)
        self.assertEqual(amount(-333.333), x1)
        x1 -= 456.456
        self.assertEqual(amount(-789.789), x1)
        x1 -= 456L
        self.assertEqual(amount(-1245.789), x1)

        x2 = amount("123456789123456789.123456789123456789")
        y2 = amount("9872345982459.248974239578")

        self.assertEqual(amount("123446916777474329.874482549545456789"), x2 - y2)
        self.assertEqual(amount("-123446916777474329.874482549545456789"), y2 - x2)

    def testIntegerMultiplication(self):
        x1 = amount(123L)
        y1 = amount(456L)

        self.assertEqual(amount(0L), x1 * 0L)
        self.assertEqual(amount(0L), amount(0L) * x1)
        self.assertEqual(amount(0L), 0L * x1)
        self.assertEqual(x1, x1 * 1L)
        self.assertEqual(x1, amount(1L) * x1)
        self.assertEqual(x1, 1L * x1)
        self.assertEqual(- x1, x1 * -1L)
        self.assertEqual(- x1, amount(-1L) * x1)
        self.assertEqual(- x1, -1L * x1)
        self.assertEqual(amount(56088L), x1 * y1)
        self.assertEqual(amount(56088L), y1 * x1)
        self.assertEqual(amount(56088L), x1 * 456L)
        self.assertEqual(amount(56088L), amount(456L) * x1)
        self.assertEqual(amount(56088L), 456L * x1)

        x1 *= amount(123L)
        self.assertEqual(amount(15129L), x1)
        x1 *= 123L
        self.assertEqual(amount(1860867L), x1)

        x3 = amount(True)
        y3 = amount(True)

        self.assertEqual(amount(True), x3 * y3)

        x4 = amount("123456789123456789123456789")

        self.assertEqual(amount("15241578780673678546105778281054720515622620750190521"),
                    x4 * x4)

    def testFractionalMultiplication(self):
        x1 = amount(123.123)
        y1 = amount(456.456)

        self.assertEqual(amount(0L), x1 * 0L)
        self.assertEqual(amount(0L), amount(0L) * x1)
        self.assertEqual(amount(0L), 0L * x1)
        self.assertEqual(x1, x1 * 1L)
        self.assertEqual(x1, amount(1L) * x1)
        self.assertEqual(x1, 1L * x1)
        self.assertEqual(- x1, x1 * -1L)
        self.assertEqual(- x1, amount(-1L) * x1)
        self.assertEqual(- x1, -1L * x1)
        self.assertEqual(amount("56200.232088"), x1 * y1)
        self.assertEqual(amount("56200.232088"), y1 * x1)
        self.assertEqual(amount("56200.232088"), x1 * 456.456)
        self.assertEqual(amount("56200.232088"), amount(456.456) * x1)
        self.assertEqual(amount("56200.232088"), 456.456 * x1)

        x1 *= amount(123.123)
        self.assertEqual(amount("15159.273129"), x1)
        x1 *= 123.123
        self.assertEqual(amount("1866455.185461867"), x1)
        x1 *= 123L
        self.assertEqual(amount("229573987.811809641"), x1)

        x2 = amount("123456789123456789.123456789123456789")

        self.assertEqual(amount("15241578780673678546105778311537878.046486820281054720515622620750190521"),
                    x2 * x2)

    def divideByZero(self, amt):
        return amt / 0

    def testIntegerDivision(self):
        x1 = amount(123L)
        y1 = amount(456L)

        self.assertRaises(exceptions.ArithmeticError, self.divideByZero, x1)
        self.assertEqual(amount(0L), amount(0L) / x1)
        self.assertEqual(amount(0L), 0L / x1)
        self.assertEqual(x1, x1 / 1L)
        self.assertEqual(amount("0.008130"), amount(1L) / x1)
        self.assertEqual(amount("0.008130"), 1L / x1)
        self.assertEqual(- x1, x1 / -1L)
        self.assertEqual(- amount("0.008130"), amount(-1L) / x1)
        self.assertEqual(- amount("0.008130"), -1L / x1)
        self.assertEqual(amount("0.269736"), x1 / y1)
        self.assertEqual(amount("3.707317"), y1 / x1)
        self.assertEqual(amount("0.269736"), x1 / 456L)
        self.assertEqual(amount("3.707317"), amount(456L) / x1)
        self.assertEqual(amount("3.707317"), 456L / x1)

        x1 /= amount(456L)
        self.assertEqual(amount("0.269736"), x1)
        x1 /= 456L
        self.assertEqual(amount("0.000591526315789473"), x1)

        x4 = amount("123456789123456789123456789")
        y4 = amount("56")

        self.assertEqual(amount(1L), x4 / x4)
        self.assertEqual(amount("2204585520061728377204585.517857"), x4 / y4)

    def testFractionalDivision(self):
        x1 = amount(123.123)
        y1 = amount(456.456)

        self.assertRaises(exceptions.ArithmeticError, self.divideByZero, x1)
        self.assertEqual(amount("0.008121"), amount(1.0) / x1)
        self.assertEqual(amount("0.008121"), 1.0 / x1)
        self.assertEqual(x1, x1 / 1.0)
        self.assertEqual(amount("0.008121"), amount(1.0) / x1)
        self.assertEqual(amount("0.008121"), 1.0 / x1)
        self.assertEqual(- x1, x1 / -1.0)
        self.assertEqual(- amount("0.008121"), amount(-1.0) / x1)
        self.assertEqual(- amount("0.008121"), -1.0 / x1)
        self.assertEqual(amount("0.269736842105"), x1 / y1)
        self.assertEqual(amount("3.707317073170"), y1 / x1)
        self.assertEqual(amount("0.269736842105"), x1 / 456.456)
        self.assertEqual(amount("3.707317073170"), amount(456.456) / x1)
        self.assertEqual(amount("3.707317073170"), 456.456 / x1)

        x1 /= amount(456.456)
        self.assertEqual(amount("0.269736842105"), x1)
        x1 /= 456.456
        self.assertEqual(amount("0.0005909372252856792330476541"), x1)
        x1 /= 456L
        self.assertEqual(amount("0.00000129591496773175270405187302631578947368421052631578947368421"), x1)

        x4 = amount("1234567891234567.89123456789")
        y4 = amount("56.789")

        self.assertEqual(amount(1.0), x4 / x4)
        self.assertEqual(amount("21739560323910.7554497273748437197344556164"),
                         x4 / y4)

def suite():
    return unittest.TestLoader().loadTestsFromTestCase(BasicAmountTestCase)

if __name__ == '__main__':
    unittest.main()