code: def test_fig_argument():
      """Test figure argument is used correctly"""
  
      fig = plt.figure()
      output_fig, ax = tape_mpl(tape1, fontsize=20, fig=fig)
  
      assert ax.get_figure() == fig
      assert output_fig == fig
      plt.close()
  
imports: import pytest
  import pennylane
  from pennylane.drawer import tape_mpl
  from pennylane.tape import QuantumScript
label_desc: Use image node_id 3 for calling a global function with example usage: test_fig_argument() without return types