def _base_unit(self):
p = self.decimal_point()
- assert p in [5,8]
- return "BTC" if p == 8 else "mBTC"
+ assert p in [2, 5, 8]
+ if p == 8:
+ return 'BTC'
+ if p == 5:
+ return 'mBTC'
+ if p == 2:
+ return 'bits'
+ raise Exception('Unknown base unit')
def get_amount(self):
try: