Add Balance class tests

1 files changed, 4 insertions, 7 deletions

diff --git a/portfolio.py b/portfolio.py
index a4fbf94..1d8bfd5 100644
--- a/portfolio.py
+++ b/portfolio.py
@@ -221,6 +221,10 @@ class Balance:
         self.used = Amount(currency, used_value)
 
     @classmethod
+    def from_hash(cls, currency, hash_):
+        return cls(currency, hash_["total"], hash_["free"], hash_["used"])
+
+    @classmethod
     def in_currency(cls, other_currency, market, action="average", type="total"):
         amounts = {}
         for currency in cls.known_balances:
@@ -235,10 +239,6 @@ class Balance:
         return cls.known_balances.keys()
 
     @classmethod
-    def from_hash(cls, currency, hash_):
-        return cls(currency, hash_["total"], hash_["free"], hash_["used"])
-
-    @classmethod
     def _fill_balances(cls, hash_):
         for key in hash_:
             if key in ["info", "free", "used", "total"]:
@@ -270,9 +270,6 @@ class Balance:
         new_repartition = cls.dispatch_assets(total_base_value)
         Trade.compute_trades(values_in_base, new_repartition, market=market)
 
-    def __int__(self):
-        return int(self.total)
-
     def __repr__(self):
         return "Balance({} [{}/{}/{}])".format(self.currency, str(self.free), str(self.used), str(self.total))