forex-python/tests/test.py at 604c228eb0a663619cd6034c68b3e4c5d3eca521 · MicroPyramid/forex-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
import datetime
from decimal import Decimal
from unittest import TestCase
from forex_python.converter import (get_rates, get_rate, convert, get_symbol,
                                    get_currency_name, RatesNotAvailableError,
                                    CurrencyRates, DecimalFloatMismatchError)


class TestGetRates(TestCase):
    """
    Test get_rates with valid(ex: USD) and invalid(ex: XYZ) currency code
    """

    def test_get_rates_valid_code(self):
        all_rates = get_rates('USD')

        # Check if return value of get_rates dictionary
        self.assertTrue(isinstance(all_rates, dict))

        # Test at least one rate value returned
        self.assertTrue(len(all_rates.keys()))

        # Test one rate in returned dict is float value
        self.assertTrue(isinstance(all_rates.get('INR'), float))

    def test_get_rates_with_date(self):
        date_obj = datetime.datetime.strptime('2010-05-10', "%Y-%m-%d").date()
        all_rates = get_rates('USD', date_obj)

        # Check if return value of get_rates dictionary
        self.assertTrue(isinstance(all_rates, dict))

        # Test at least one rate value returned
        self.assertTrue(len(all_rates.keys()))

        # Test one rate in returned dict is float value
        self.assertTrue(isinstance(all_rates.get('INR'), float))

    def test_get_rates_invalid_code(self):
        self.assertRaises(RatesNotAvailableError, get_rates, 'XYZ')

    # def test_get_rates_in_future(self):
    #     future = datetime.date.today() + datetime.timedelta(days=1)
    #     self.assertRaises(RatesNotAvailableError, get_rates, 'USD', future)


class TestGetRate(TestCase):
    """
    Test get_rate function using valid and invalid currency codes
    """

    def test_get_rate_with_valid_codes(self):
        rate = get_rate('USD', 'INR')

        # check if return value is float
        self.assertTrue(isinstance(rate, float))

    def test_get_rate_with_valid_codes_same_currency(self):
        rate = get_rate('USD', 'USD')
        # rate should be 1.
        self.assertEqual(1., rate)

    def test_get_rate_with_date(self):
        date_obj = datetime.datetime.strptime('2010-05-10', "%Y-%m-%d").date()
        rate = get_rate('USD', 'INR', date_obj)

        # check if return value is float
        self.assertTrue(isinstance(rate, float))

    def test_get_rate_with_invalid_codes(self):
        # raise exception for invalid currency codes
        self.assertRaises(RatesNotAvailableError, get_rate, 'ABCD', 'XYZ')

    # def test_get_rates_in_future(self):
    #     future = datetime.date.today() + datetime.timedelta(days=1)
    #     self.assertRaises(
    #         RatesNotAvailableError, get_rate, 'EUR', 'USD', future)


class TestAmountConvert(TestCase):
    """
    test amount conversion from one currency to other
    """

    def test_amount_convert_valid_currency(self):
        amount = convert('USD', 'INR', 10)

        # test if amount returned in float
        self.assertTrue(isinstance(amount, float))

    def test_amount_convert_valid_currency_same_currency(self):
        amount = convert('USD', 'USD', 10)
        self.assertEqual(amount, float(10))


    def test_amount_convert_date(self):
        date_obj = datetime.datetime.strptime('2010-05-10', "%Y-%m-%d").date()
        amount = convert('USD', 'INR', 10, date_obj)

        # test if amount returned in float
        self.assertTrue(isinstance(amount, float))

    def test_amount_convert_invalid_currency(self):
        # test if amount returned in float
        self.assertRaises(RatesNotAvailableError, convert, 'ABC', 'XYZ', 10)


class TestForceDecimalAmountConvert(TestCase):
    """
    Test the force_decimal=True type enforcing
    """

    def setUp(self):
        self.c = CurrencyRates(force_decimal=True)

    def test_amount_decimal_convert(self):
        amount = self.c.convert('USD', 'INR', Decimal('10.45'))
        self.assertTrue(isinstance(amount, Decimal))

    def test_amount_decimal_convert_same_currency(self):
        amount = self.c.convert('USD', 'USD', Decimal('10.45'))
        self.assertEqual(amount, Decimal('10.45'))

    def test_amount_decimal_convert_date(self):
        date_obj = datetime.datetime.strptime('2010-05-10', "%Y-%m-%d").date()
        amount = self.c.convert('USD', 'INR', Decimal('10.45'), date_obj)

        self.assertTrue(isinstance(amount, Decimal))

    def test_amount_decimal_invalid_type(self):
        self.assertRaises(DecimalFloatMismatchError, self.c.convert, 'USD', 'INR', 10.45)

    def test_decimal_get_rates_valid_code(self):
        all_rates = self.c.get_rates('USD')
        # Check if return value of get_rates dictionary
        self.assertTrue(isinstance(all_rates, dict))
        # Test at least one rate value returned
        self.assertTrue(len(all_rates.keys()))
        # Test one rate in returned dict is now a Decimal
        self.assertTrue(isinstance(all_rates.get('INR'), Decimal))

    def test_decimal_get_rates_with_date(self):
        date_obj = datetime.datetime.strptime('2010-05-10', "%Y-%m-%d").date()
        all_rates = self.c.get_rates('USD', date_obj)
        # Check if return value of get_rates dictionary
        self.assertTrue(isinstance(all_rates, dict))
        # Test at least one rate value returned
        self.assertTrue(len(all_rates.keys()))
        # Test one rate in returned dict is now a Decimal
        self.assertTrue(isinstance(all_rates.get('INR'), Decimal))

    def test_decimal_get_rates_invalid_code(self):
        self.assertRaises(RatesNotAvailableError, self.c.get_rates, 'XYZ')

    def test_decimal_get_rate_with_valid_codes(self):
        rate = self.c.get_rate('USD', 'INR')
        # check if return value is Decimal
        self.assertTrue(isinstance(rate, Decimal))

    def test_decimal_get_rate_with_valid_same_codes(self):
        rate = self.c.get_rate('USD', 'USD')
        # check if return value is Decimal
        self.assertEqual(rate, Decimal(1))


    def test_decimal_get_rate_with_date(self):
        date_obj = datetime.datetime.strptime('2010-05-10', "%Y-%m-%d").date()
        rate = self.c.get_rate('USD', 'INR', date_obj)
        # check if return value is Decimal
        self.assertTrue(isinstance(rate, Decimal))

    def test_decimal_get_rate_with_invalid_codes(self):
        # raise exception for invalid currency codes
        self.assertRaises(RatesNotAvailableError, self.c.get_rate, 'ABCD', 'XYZ')


class TestCurrencySymbol(TestCase):
    """
    test currency symbols from currency codes
    """

    def test_with_valid_currency_code(self):
        self.assertEqual(str(get_symbol('USD')), '$')

    def test_with_invalid_currency_code(self):
        self.assertFalse(get_symbol('XYZ'))


class TestCurrencyName(TestCase):
    """
    test currency name from currency codes
    """

    def test_with_valid_currency_code(self):
        self.assertEqual(str(get_currency_name('USD')), 'United States dollar')

    def test_with_invalid_currency_code(self):
        self.assertFalse(get_currency_name('XYZ'))