code: def test_fig_argument():
      """Tests figure argument is used correcly"""
  
      fig = plt.figure()
      output_fig, ax = qml.draw_mpl(circuit1, fig=fig)(1.23, 2.34)
      assert ax.get_figure() == fig
      assert output_fig == fig
  
imports: import pytest
  import pennylane
label_desc: Use image node_id 4 for calling a global function with example usage: test_fig_argument() without return types