diff --git a/tests/test_reports_balance.py b/tests/test_reports_balance.py index 249a288d9890209e8265085d419725db5e8f130c..07e238662870c712fc76a39a84f210ac8aa691c1 100644 --- a/tests/test_reports_balance.py +++ b/tests/test_reports_balance.py @@ -133,6 +133,23 @@ def test_neg(balance_map_kwargs): for key in balance_map_kwargs: assert actual[key] == -amounts[key] +@pytest.mark.parametrize('kwargs1,kwargs2,expected', [ + ({}, {}, True), + ({}, {'USD': 0}, True), + ({}, {'EUR': 1}, False), + ({'USD': 1}, {'EUR': 1}, False), + ({'USD': 1}, {'USD': '1.0'}, True), + ({'USD': 1}, {'USD': '1.0', 'EUR': '2.0'}, False), + ({'USD': 1, 'BRL': '2.0'}, {'USD': '1.0', 'EUR': '2.0'}, False), + ({'USD': 1, 'EUR': 2, 'BRL': '3.0'}, {'USD': '1.0', 'EUR': '2.0'}, False), + ({'USD': 1, 'EUR': 2}, {'USD': '1.0', 'EUR': '2.0'}, True), +]) +def test_eq(kwargs1, kwargs2, expected): + bal1 = core.Balance(testutil.balance_map(**kwargs1)) + bal2 = core.Balance(testutil.balance_map(**kwargs2)) + actual = bal1 == bal2 + assert actual == expected + @pytest.mark.parametrize('balance_map_kwargs,expected', [ ({}, "Zero balance"), ({'JPY': 0, 'BRL': 0}, "Zero balance"),