def test_save_load_updates_csv(self):
updates = BalanceUpdates.from_csv(self.updates_csv_path)
updates.to_csv("tmp.csv")
updates_from_save = BalanceUpdates.from_csv("tmp.csv")
pd.testing.assert_frame_equal(updates.df, updates_from_save.df)
os.remove("tmp.csv")
self.assertTrue(True)
def test_get_daily_balances_with_trans(self):
bal_trans = BalanceTransfers.from_dict({
"from_account_key": ["1", "2", "1", "2", "1", "2"],
"to_account_key": ["2", "1", "3", "1", "2", "2"],
"amount": [10.0, 5.0, 4.0, 15.0, 5.0, 100],
"date": ["2019-01-03", "2019-01-03", "2019-01-03", "2019-01-04", "2019-01-05", "2019-01-01"],
})
balance_updates = BalanceUpdates.from_dict({
"account_key": ["1", "1"],
"balance": [100.0, 200.0],
"date": ["2019-01-01", "2019-01-05"],
})
hist = MoneyFrame.from_updates_and_transfers(balance_transfers=bal_trans,
balance_updates=balance_updates, account_key="1")
daily_bals = hist.get_daily_balance()
print(daily_bals)
mult = 0.18904171541952763 + 1.0
exp = pd.DataFrame({
"date": pd.to_datetime(["2019-01-01", "2019-01-02", "2019-01-03", "2019-01-04", "2019-01-05"]),
"balance": [
100.0,
100.0 * mult,
100.0 * mult * mult - 9.0,
(100.0 * mult * mult - 9.0) * mult + 15.0,
((100.0 * mult * mult - 9.0) * mult + 15.0) * mult - 5.0
],
}).set_index("date")
np.testing.assert_equal(exp.index.values, daily_bals.index.values)
np.testing.assert_array_almost_equal(exp["balance"].values, daily_bals.values)
def test_get_acc_updates(self):
balance_updates = BalanceUpdates.from_dict({
field_names.ACCOUNT_KEY: ["1", "2", "1"],
field_names.BALANCE: [100.0, 200.0, 200.0],
field_names.DATE: ["2019-01-01", "2019-01-01", "2019-01-03"],
})
df = balance_updates.get_acc_updates(
account_key="1").to_frame().reset_index()
print(df)
exp = pd.DataFrame({
field_names.BALANCE: [100.0, 200.0],
field_names.DATE:
pd.to_datetime(["2019-01-01", "2019-01-03"]),
})
self.assertTrue(compare_pd_df(df, exp, sort=False))
df = balance_updates.get_acc_updates(
account_key="1").to_frame().reset_index()
exp = pd.DataFrame({
field_names.BALANCE: [200.0, 100.0],
field_names.DATE:
pd.to_datetime(["2019-01-03", "2019-01-01"]),
})
self.assertTrue(compare_pd_df(df, exp, sort=True))
def test_daily_interest_amounts(self):
bal_trans = BalanceTransfers.from_dict({
"from_account_key": ["0", "0"],
"to_account_key": ["1", "1"],
"amount": [90.0, 90.0],
"date": ["2019-01-03", "2018-12-30"],
})
balance_updates = BalanceUpdates.from_dict({
"account_key": ["1", "1"],
"balance": [100.0, 200.0],
"date": ["2019-01-01", "2019-01-05"],
})
exp = pd.DataFrame({
"date": pd.to_datetime(["2018-12-29", "2018-12-30", "2018-12-31", "2019-01-01", "2019-01-02",
"2019-01-03", "2019-01-04", "2019-01-05"]),
"interest": [0.0, 0.0, 4.868330, 10.0 - 4.868330, 1.689833, 193.408221 - 90.0 - 101.689833,
196.676496 - 193.408221,
200.0 - 196.676496],
}).set_index("date")["interest"]
hist = MoneyFrame.from_updates_and_transfers(balance_transfers=bal_trans,
balance_updates=balance_updates, account_key="1")
print(hist.get_daily_balance())
print(hist.get_daily_transfers())
daily_interest = hist.get_daily_interest()
print(daily_interest)
np.testing.assert_equal(exp.index.values, daily_interest.index.values)
np.testing.assert_array_almost_equal(exp.values, daily_interest.values, 4)
def test_get_daily_balances(self):
bal_trans = BalanceTransfers.from_dict({
"from_account_key": [],
"to_account_key": [],
"amount": [],
"date": [],
})
balance_updates = BalanceUpdates.from_dict({
"account_key": ["1", "1", "1", "1", "1"],
"balance": [100.0, 200.0, 220.0, 220.0, 200.0],
"date": ["2019-01-01", "2019-01-03", "2019-01-04", "2019-01-06", "2019-01-08"],
})
hist = MoneyFrame.from_updates_and_transfers(balance_transfers=bal_trans,
balance_updates=balance_updates, account_key="1")
daily_bals = hist.get_daily_balance()
exp = pd.DataFrame({
"date": pd.to_datetime(["2019-01-01", "2019-01-02", "2019-01-03", "2019-01-04", "2019-01-05", "2019-01-06",
"2019-01-07", "2019-01-08"]),
"balance": [100.0, 141.421356, 200.0, 220.0, 220.0, 220.0, 209.761770, 200.0],
}).set_index("date")
np.testing.assert_equal(exp.index.values, daily_bals.index.values)
np.testing.assert_array_almost_equal(exp["balance"].values, daily_bals.values)
def test_equals(self):
balance_updates1 = BalanceUpdates.from_dict({
field_names.ACCOUNT_KEY: ["1", "2", "1"],
field_names.BALANCE: [100.0, 200.0, 200.0],
field_names.DATE: ["2019-01-01", "2019-01-01", "2019-01-03"],
})
balance_updates2 = BalanceUpdates.from_dict({
field_names.ACCOUNT_KEY: ["1", "2", "1"],
field_names.BALANCE: [100.0, 200.0, 200.0],
field_names.DATE: ["2019-01-01", "2019-01-01", "2019-01-03"],
})
balance_updates3 = BalanceUpdates.from_dict({
field_names.ACCOUNT_KEY: ["1", "1", "2"],
field_names.BALANCE: [100.0, 200.0, 200.0],
field_names.DATE: ["2019-01-01", "2019-01-03", "2019-01-01"],
})
balance_updates4 = BalanceUpdates.from_dict({
field_names.ACCOUNT_KEY: ["1", "2", "2"],
field_names.BALANCE: [100.0, 200.0, 200.0],
field_names.DATE: ["2019-01-01", "2019-01-03", "2019-01-01"],
})
self.assertTrue(balance_updates1.equals(balance_updates2))
self.assertTrue(balance_updates1.equals(balance_updates3))
self.assertFalse(balance_updates1.equals(balance_updates4))
def test_daily_summary(self):
bal_trans = BalanceTransfers.from_dict({
"from_account_key": ["0", "0"],
"to_account_key": ["1", "1"],
"amount": [90.0, 90.0],
"date": ["2019-01-03", "2018-12-30"],
})
balance_updates = BalanceUpdates.from_dict({
"account_key": ["1", "1"],
"balance": [100.0, 200.0],
"date": ["2019-01-01", "2019-01-05"],
})
hist = MoneyFrame.from_updates_and_transfers(balance_transfers=bal_trans, balance_updates=balance_updates,
account_key="1")
print(hist.to_df())
def test_get_daily_balances_edge_case_1(self):
bal_trans = BalanceTransfers.from_dict({
"from_account_key": ["0", "0"],
"to_account_key": ["1", "1"],
"amount": [90.0, 90.0],
"date": ["2019-01-03", "2018-12-30"],
})
balance_updates = BalanceUpdates.from_dict({
"account_key": [],
"balance": [],
"date": [],
})
hist = MoneyFrame.from_updates_and_transfers(balance_transfers=bal_trans,
balance_updates=balance_updates, account_key="1")
daily_bals = hist.get_daily_balance()
exp = pd.DataFrame({
"date": pd.to_datetime(
["2018-12-29", "2018-12-30", "2018-12-31", "2019-01-01", "2019-01-02", "2019-01-03"]),
"balance": [0.0, 90.0, 90.0, 90.0, 90.0, 180.0],
}).set_index("date")
np.testing.assert_equal(exp.index.values, daily_bals.index.values)
np.testing.assert_array_almost_equal(exp["balance"].values, daily_bals.values)
def test_get_daily_balances_with_early_trans(self):
bal_trans = BalanceTransfers.from_dict({
"from_account_key": ["0", "0"],
"to_account_key": ["1", "1"],
"amount": [90.0, 90.0],
"date": ["2019-01-03", "2018-12-30"],
})
balance_updates = BalanceUpdates.from_dict({
"account_key": ["1", "1"],
"balance": [100.0, 200.0],
"date": ["2019-01-01", "2019-01-05"],
})
hist = MoneyFrame.from_updates_and_transfers(balance_transfers=bal_trans,
balance_updates=balance_updates, account_key="1")
daily_bals = hist.get_daily_balance()
mult = 1.05409255338945984
mult2 = 1.01689832725085294
exp = pd.DataFrame({
"date": pd.to_datetime(["2018-12-29", "2018-12-30", "2018-12-31", "2019-01-01", "2019-01-02",
"2019-01-03", "2019-01-04", "2019-01-05"]),
"balance": [
0.0, # 29
90.0, # 30
90.0 * mult, # 31
90.0 * mult * mult, # 01
100.0 * mult2, # 02
100.0 * mult2 * mult2 + 90.0, # 03
(100.0 * mult2 * mult2 + 90.0) * mult2, # 04
(100.0 * mult2 * mult2 + 90.0) * mult2 * mult2, # 05
],
}).set_index("date")
np.testing.assert_equal(exp.index.values, daily_bals.index.values)
np.testing.assert_array_almost_equal(exp["balance"].values, daily_bals.values)
def test_save_load_updates_excel(self):
accounts_1 = BalanceUpdates.from_excel(self.excel_path,
"balance_updates")
accounts_2 = BalanceUpdates.from_csv(self.updates_csv_path)
pd.testing.assert_frame_equal(accounts_1.df, accounts_2.df)