code: def test_total_balance(portfolio_locked):
      assert portfolio_locked.total_balance(USD) == 10000
      assert portfolio_locked.total_balance(BTC) == 1
      assert portfolio_locked.total_balance(ETH) == 10
      assert portfolio_locked.total_balance(XRP) == 5000
  
imports: import pytest
  import unittest.mock
  from tensortrade.core.base import Clock
  from tensortrade.oms.wallets import Wallet, Portfolio
  from tensortrade.oms.instruments import USD, BTC, ETH, XRP, BCH
label_desc: Use image node_id 13 for calling a global function with example usage: test_total_balance(portfolio_locked) without return types