code: def test_balance(portfolio_locked):
      assert portfolio_locked.balance(USD) == 9850
      assert portfolio_locked.balance(BTC) == 0.25
      assert portfolio_locked.balance(ETH) == 3
      assert portfolio_locked.balance(XRP) == 4738
  
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 11 for calling a global function with example usage: test_balance(portfolio_locked) without return types