From 7eb9cb36c2ef01097fe8e99d40dd17308abdfe37 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Isma=C3=ABl=20Bouya?= Date: Mon, 19 Feb 2018 14:06:45 +0100 Subject: [PATCH] Allow dispatch assets to receive liquidity --- store.py | 4 ++-- test.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/store.py b/store.py index 841a0fc..abbe5ee 100644 --- a/store.py +++ b/store.py @@ -27,9 +27,9 @@ class BalanceStore: cls.all[currency] = portfolio.Balance(currency, balance) @classmethod - def dispatch_assets(cls, amount, repartition=None): + def dispatch_assets(cls, amount, liquidity="medium", repartition=None): if repartition is None: - repartition = portfolio.Portfolio.repartition() + repartition = portfolio.Portfolio.repartition(liquidity=liquidity) sum_ratio = sum([v[0] for k, v in repartition.items()]) amounts = {} for currency, (ptt, trade_type) in repartition.items(): diff --git a/test.py b/test.py index 8641130..d58b203 100644 --- a/test.py +++ b/test.py @@ -1247,6 +1247,7 @@ class BalanceStoreTest(WebMockTestCase): } amounts = portfolio.BalanceStore.dispatch_assets(portfolio.Amount("BTC", "11.1")) + repartition.assert_called_with(liquidity="medium") self.assertIn("XEM", portfolio.BalanceStore.currencies()) self.assertEqual(D("2.6"), amounts["BTC"].value) self.assertEqual(D("7.5"), amounts["XEM"].value) -- 2.41.0