diff options
Diffstat (limited to 'tests/models/test_currency.py')
| -rw-r--r-- | tests/models/test_currency.py | 410 |
1 files changed, 410 insertions, 0 deletions
diff --git a/tests/models/test_currency.py b/tests/models/test_currency.py new file mode 100644 index 0000000..40cff88 --- /dev/null +++ b/tests/models/test_currency.py @@ -0,0 +1,410 @@ +"""These were taken from the wxet project's first use of this idea. Not all +functionality is the same, but pasting here so the tests are in the +correct spot... +""" + +from decimal import Decimal +from random import randint + +import pytest + + +class TestUSDCentModel: + + def test_construct_int(self): + from generalresearch.currency import USDCent + + for i in range(100): + int_val = randint(0, 999_999) + instance = USDCent(int_val) + assert int_val == instance + + def test_construct_float(self): + from generalresearch.currency import USDCent + + with pytest.warns(expected_warning=Warning) as record: + float_val: float = 10.6789 + instance = USDCent(float_val) + + assert len(record) == 1 + assert "USDCent init with a float. Rounding behavior may be unexpected" in str( + record[0].message + ) + assert instance == USDCent(10) + assert instance == 10 + + def test_construct_decimal(self): + from generalresearch.currency import USDCent + + with pytest.warns(expected_warning=Warning) as record: + decimal_val: Decimal = Decimal("10.0") + instance = USDCent(decimal_val) + + assert len(record) == 1 + assert ( + "USDCent init with a Decimal. Rounding behavior may be unexpected" + in str(record[0].message) + ) + + assert instance == USDCent(10) + assert instance == 10 + + # Now with rounding + with pytest.warns(Warning) as record: + decimal_val: Decimal = Decimal("10.6789") + instance = USDCent(decimal_val) + + assert len(record) == 1 + assert ( + "USDCent init with a Decimal. Rounding behavior may be unexpected" + in str(record[0].message) + ) + + assert instance == USDCent(10) + assert instance == 10 + + def test_construct_negative(self): + from generalresearch.currency import USDCent + + with pytest.raises(expected_exception=ValueError) as cm: + USDCent(-1) + assert "USDCent not be less than zero" in str(cm.value) + + def test_operation_add(self): + from generalresearch.currency import USDCent + + for i in range(100): + int_val1 = randint(0, 999_999) + int_val2 = randint(0, 999_999) + + instance1 = USDCent(int_val1) + instance2 = USDCent(int_val2) + + assert int_val1 + int_val2 == instance1 + instance2 + + def test_operation_subtract(self): + from generalresearch.currency import USDCent + + for i in range(100): + int_val1 = randint(500_000, 999_999) + int_val2 = randint(0, 499_999) + + instance1 = USDCent(int_val1) + instance2 = USDCent(int_val2) + + assert int_val1 - int_val2 == instance1 - instance2 + + def test_operation_subtract_to_neg(self): + from generalresearch.currency import USDCent + + for i in range(100): + int_val = randint(0, 999_999) + instance = USDCent(int_val) + + with pytest.raises(expected_exception=ValueError) as cm: + instance - USDCent(1_000_000) + + assert "USDCent not be less than zero" in str(cm.value) + + def test_operation_multiply(self): + from generalresearch.currency import USDCent + + for i in range(100): + int_val1 = randint(0, 999_999) + int_val2 = randint(0, 999_999) + + instance1 = USDCent(int_val1) + instance2 = USDCent(int_val2) + + assert int_val1 * int_val2 == instance1 * instance2 + + def test_operation_div(self): + from generalresearch.currency import USDCent + + with pytest.raises(ValueError) as cm: + USDCent(10) / 2 + assert "Division not allowed for USDCent" in str(cm.value) + + def test_operation_result_type(self): + from generalresearch.currency import USDCent + + int_val = randint(1, 999_999) + instance = USDCent(int_val) + + res_add = instance + USDCent(1) + assert isinstance(res_add, USDCent) + + res_sub = instance - USDCent(1) + assert isinstance(res_sub, USDCent) + + res_multipy = instance * USDCent(2) + assert isinstance(res_multipy, USDCent) + + def test_operation_partner_add(self): + from generalresearch.currency import USDCent + + int_val = randint(1, 999_999) + instance = USDCent(int_val) + + with pytest.raises(expected_exception=AssertionError): + instance + 0.10 + + with pytest.raises(expected_exception=AssertionError): + instance + Decimal(".10") + + with pytest.raises(expected_exception=AssertionError): + instance + "9.9" + + with pytest.raises(expected_exception=AssertionError): + instance + True + + def test_abs(self): + from generalresearch.currency import USDCent + + for i in range(100): + int_val = abs(randint(0, 999_999)) + instance = abs(USDCent(int_val)) + + assert int_val == instance + + def test_str(self): + from generalresearch.currency import USDCent + + for i in range(100): + int_val = randint(0, 999_999) + instance = USDCent(int_val) + + assert str(int_val) == str(instance) + + def test_operation_result_type_unsupported(self): + """There is no correct answer here, but we at least want to make sure + that a USDCent is returned + """ + from generalresearch.currency import USDCent + + res = USDCent(10) // 1.2 + assert not isinstance(res, USDCent) + assert isinstance(res, float) + + res = USDCent(10) % 1 + assert not isinstance(res, USDCent) + assert isinstance(res, int) + + res = pow(USDCent(10), 2) + assert not isinstance(res, USDCent) + assert isinstance(res, int) + + res = pow(USDCent(10), USDCent(2)) + assert not isinstance(res, USDCent) + assert isinstance(res, int) + + res = float(USDCent(10)) + assert not isinstance(res, USDCent) + assert isinstance(res, float) + + +class TestUSDMillModel: + + def test_construct_int(self): + from generalresearch.currency import USDMill + + for i in range(100): + int_val = randint(0, 999_999) + instance = USDMill(int_val) + assert int_val == instance + + def test_construct_float(self): + from generalresearch.currency import USDMill + + with pytest.warns(expected_warning=Warning) as record: + float_val: float = 10.6789 + instance = USDMill(float_val) + + assert len(record) == 1 + assert "USDMill init with a float. Rounding behavior may be unexpected" in str( + record[0].message + ) + assert instance == USDMill(10) + assert instance == 10 + + def test_construct_decimal(self): + from generalresearch.currency import USDMill + + with pytest.warns(expected_warning=Warning) as record: + decimal_val: Decimal = Decimal("10.0") + instance = USDMill(decimal_val) + + assert len(record) == 1 + assert ( + "USDMill init with a Decimal. Rounding behavior may be unexpected" + in str(record[0].message) + ) + + assert instance == USDMill(10) + assert instance == 10 + + # Now with rounding + with pytest.warns(expected_warning=Warning) as record: + decimal_val: Decimal = Decimal("10.6789") + instance = USDMill(decimal_val) + + assert len(record) == 1 + assert ( + "USDMill init with a Decimal. Rounding behavior may be unexpected" + in str(record[0].message) + ) + + assert instance == USDMill(10) + assert instance == 10 + + def test_construct_negative(self): + from generalresearch.currency import USDMill + + with pytest.raises(expected_exception=ValueError) as cm: + USDMill(-1) + assert "USDMill not be less than zero" in str(cm.value) + + def test_operation_add(self): + from generalresearch.currency import USDMill + + for i in range(100): + int_val1 = randint(0, 999_999) + int_val2 = randint(0, 999_999) + + instance1 = USDMill(int_val1) + instance2 = USDMill(int_val2) + + assert int_val1 + int_val2 == instance1 + instance2 + + def test_operation_subtract(self): + from generalresearch.currency import USDMill + + for i in range(100): + int_val1 = randint(500_000, 999_999) + int_val2 = randint(0, 499_999) + + instance1 = USDMill(int_val1) + instance2 = USDMill(int_val2) + + assert int_val1 - int_val2 == instance1 - instance2 + + def test_operation_subtract_to_neg(self): + from generalresearch.currency import USDMill + + for i in range(100): + int_val = randint(0, 999_999) + instance = USDMill(int_val) + + with pytest.raises(expected_exception=ValueError) as cm: + instance - USDMill(1_000_000) + + assert "USDMill not be less than zero" in str(cm.value) + + def test_operation_multiply(self): + from generalresearch.currency import USDMill + + for i in range(100): + int_val1 = randint(0, 999_999) + int_val2 = randint(0, 999_999) + + instance1 = USDMill(int_val1) + instance2 = USDMill(int_val2) + + assert int_val1 * int_val2 == instance1 * instance2 + + def test_operation_div(self): + from generalresearch.currency import USDMill + + with pytest.raises(ValueError) as cm: + USDMill(10) / 2 + assert "Division not allowed for USDMill" in str(cm.value) + + def test_operation_result_type(self): + from generalresearch.currency import USDMill + + int_val = randint(1, 999_999) + instance = USDMill(int_val) + + res_add = instance + USDMill(1) + assert isinstance(res_add, USDMill) + + res_sub = instance - USDMill(1) + assert isinstance(res_sub, USDMill) + + res_multipy = instance * USDMill(2) + assert isinstance(res_multipy, USDMill) + + def test_operation_partner_add(self): + from generalresearch.currency import USDMill + + int_val = randint(1, 999_999) + instance = USDMill(int_val) + + with pytest.raises(expected_exception=AssertionError): + instance + 0.10 + + with pytest.raises(expected_exception=AssertionError): + instance + Decimal(".10") + + with pytest.raises(expected_exception=AssertionError): + instance + "9.9" + + with pytest.raises(expected_exception=AssertionError): + instance + True + + def test_abs(self): + from generalresearch.currency import USDMill + + for i in range(100): + int_val = abs(randint(0, 999_999)) + instance = abs(USDMill(int_val)) + + assert int_val == instance + + def test_str(self): + from generalresearch.currency import USDMill + + for i in range(100): + int_val = randint(0, 999_999) + instance = USDMill(int_val) + + assert str(int_val) == str(instance) + + def test_operation_result_type_unsupported(self): + """There is no correct answer here, but we at least want to make sure + that a USDMill is returned + """ + from generalresearch.currency import USDCent, USDMill + + res = USDMill(10) // 1.2 + assert not isinstance(res, USDMill) + assert isinstance(res, float) + + res = USDMill(10) % 1 + assert not isinstance(res, USDMill) + assert isinstance(res, int) + + res = pow(USDMill(10), 2) + assert not isinstance(res, USDMill) + assert isinstance(res, int) + + res = pow(USDMill(10), USDMill(2)) + assert not isinstance(res, USDCent) + assert isinstance(res, int) + + res = float(USDMill(10)) + assert not isinstance(res, USDMill) + assert isinstance(res, float) + + +class TestNegativeFormatting: + + def test_pos(self): + from generalresearch.currency import format_usd_cent + + assert "-$987.65" == format_usd_cent(-98765) + + def test_neg(self): + from generalresearch.currency import format_usd_cent + + assert "-$123.45" == format_usd_cent(-12345) |