def test_accounting_equation_with_groups():
clear_all_definitions()
Asset.define("USD")
assets = AccountGroup.define(name="assets")
liabilities = AccountGroup.define(name="liabilities")
equity = AccountGroup.define(name="equity")
shareholder_equity = Account.define(name="shareholder_equity", asset="USD")
wells_fargo = Account.define(name="wells_fargo", asset="USD")
equipment = Account.define(name="equipment", asset="USD")
cash = Account.define(name="cash", asset="USD")
assets.add(cash)
assets.add(equipment)
equity.add(shareholder_equity)
liabilities.add(wells_fargo)
ipo_amount = AssetAmount(amount=D(100), asset="USD")
ipo_entry = DoubleEntry(src=shareholder_equity,
dest=equipment,
asset_amount=ipo_amount)
loan_amount = AssetAmount(amount=D(50), asset="USD")
loan_entry = DoubleEntry(src=wells_fargo,
dest=cash,
asset_amount=loan_amount)
with GeneralLedger.transaction_lock:
GeneralLedger.realize(ipo_entry)
GeneralLedger.realize(loan_entry)
assert_true(GeneralLedger.balance()[Asset["USD"]].amount == D(0))
assert_true(assets + liabilities +
equity == AssetAmount(asset="USD", amount=D(0)))
def test_equality():
clear_all_definitions()
Asset.define("EN")
a1 = Asset.get("EN")
a2 = Asset.get("EN")
assert_true(a1 == a2)
assert_true(a1 is a2)
def test_comparisons_and_ident():
prep_test_assets()
a = Asset.get("EN")
am0 = AssetAmount(asset=a, amount=0)
am1 = AssetAmount(asset=a, amount=1)
am2 = AssetAmount(asset=a, amount=2)
assert_true(am0 < am1)
assert_false(am0 > am1)
assert_false(am0 >= am1)
assert_true(am1 < am2)
assert_false(am1 > am2)
assert_false(am1 >= am2)
assert_true(am1 > am0)
assert_false(am1 < am0)
assert_false(am1 <= am0)
assert_true(am2 > am1)
assert_false(am2 < am1)
assert_false(am2 <= am1)
am3 = AssetAmount(asset=a, amount=0)
assert_true(am0 >= am3)
assert_true(am0 <= am3)
assert_true(am0 == am3)
assert_false(am0 != am3)
assert_false(am0 is am3)
def test_asset_amount_creation():
prep_test_assets()
a = Asset.get("EN")
AssetAmount(asset=a, amount="0")
AssetAmount(asset=a, amount=1)
AssetAmount(asset=a, amount=D(1))
AssetAmount(asset="EN", amount=2)
def prep_test_ledger():
clear_all_definitions()
Asset.define("USD")
assets = AccountGroup.define(name="assets")
liabilities = AccountGroup.define(name="liabilities")
equity = AccountGroup.define(name="equity")
shareholder_equity = Account.define(name="shareholder_equity", asset="USD")
wells_fargo = Account.define(name="wells_fargo", asset="USD")
equipment = Account.define(name="equipment", asset="USD")
cash = Account.define(name="cash", asset="USD")
assets.add(cash)
assets.add(equipment)
equity.add(shareholder_equity)
liabilities.add(wells_fargo)
def test_asset_balance_comparisons():
clear_all_definitions()
k = Asset.define("KET")
b0 = AssetBalance()
b1 = AssetBalance()
b0[k] = AssetAmount(asset=k, amount=D("123"))
b1[k] = AssetAmount(asset=k, amount=D("123"))
assert_true(AssetAmount(asset=k, amount=D("123")) == b1)
assert_true(b0 == AssetAmount(asset=k, amount=D("123")))
assert_true(b0 == b1)
def test_entries_balance_to_zero():
clear_all_definitions()
Asset.define("USD")
assets = Account.define(name="Assets", asset="USD")
liabilities = Account.define(name="Liabilities", asset="USD")
equity = Account.define(name="Equity", asset="USD")
ipo_amount = AssetAmount(amount=D(100), asset="USD")
ipo_entry = DoubleEntry(src=equity, dest=assets, asset_amount=ipo_amount)
loan_amount = AssetAmount(amount=D(50), asset="USD")
loan_entry = DoubleEntry(src=liabilities,
dest=assets,
asset_amount=loan_amount)
with GeneralLedger.transaction_lock:
GeneralLedger.realize(ipo_entry)
GeneralLedger.realize(loan_entry)
assert_true(GeneralLedger.balance()[Asset["USD"]].amount == D(0))
def test_asset_math():
prep_test_assets()
foo = Asset.get("EN")
am0 = AssetAmount(asset=foo, amount=0)
am1 = AssetAmount(asset=foo, amount=1)
am2 = AssetAmount(asset=foo, amount=2)
nosetools.eq_(am0 + am2, foo.make_amount(2))
nosetools.eq_(am2 - am1, foo.make_amount(1))
nosetools.eq_(am2 * am2 * am2, foo.make_amount(8))
nosetools.eq_(am2**(am2 + am1), foo.make_amount(8))
nosetools.eq_(am2 / am2, am1)
nosetools.eq_(am2 // am2, am1)
nosetools.eq_(am2 % am2, am0)
nosetools.eq_(-am2, -(am1 + am1))
nosetools.eq_(abs(am0 - am2), am2)
nosetools.eq_(+(am0 - am2), am2)
def test_asset_quote_str_and_repr():
prep_test_assets_and_pairs()
aq = ExchangeRate(rate=D(1.0),
base=Asset.get("EN"),
quote=Asset.get("RON"))
logger.info("Found a {0!s} made by {0!r}".format(aq))
def test_make_amount():
clear_all_definitions()
a = Asset.define("EN")
a.make_amount(D(1))
a.make_amount(10)
a.make_amount()
def cannot_doubly_define():
clear_all_definitions()
Asset.define("EN")
Asset.define("EN")
def test_asset_definition():
clear_all_definitions()
Asset.define("EN")
Asset.get("EN")
Asset["EN"]
def test_asset_balance_set():
prep_test_assets()
am = AssetAmount(asset=Asset.get("RON"), amount=1)
ab = AssetBalance()
ab[am.asset] = am
def test_cannot_instantiate_from_float_mindlessly():
prep_test_assets()
my_fun_val = math.e
a = Asset.get("EN")
Account.define(name="Donut", asset=a, amount=my_fun_val)
def test_asset_str_and_repr():
clear_all_definitions()
a = Asset.define("EN")
logger.info("Found a {0!s} made by {0!r}".format(a))
def test_asset_not_defined():
clear_all_definitions()
Asset.get("EN")
def test_asset_balance_keys_by_string():
prep_test_assets()
am = AssetAmount(asset=Asset.get("RON"), amount=1)
ab = AssetBalance()
ab["RON"] = am
ab["RON"]
def test_missing_asset_set_key_for_balance():
prep_test_assets()
am = AssetAmount(asset=Asset.get("RON"), amount=1)
ab = AssetBalance()
ab["YOWZERS"] = am
def prep_test_assets():
clear_all_definitions()
Asset.define("EN")
Asset.define("RON")
def cannot_create_asset_from_asset():
prep_test_assets()
a = Asset.get("EN")
Asset.define(a)
def cannot_compare_to_strings():
prep_test_assets()
a = Asset.define("EN")
if a != "RON":
logger.info("I think it's safe to skip this account")