code: def test_get_agg_axis(self, float_frame):
      cols = float_frame._get_agg_axis(0)
      assert cols is float_frame.columns
  
      idx = float_frame._get_agg_axis(1)
      assert idx is float_frame.index
  
      msg = r"Axis must be 0 or 1 \(got 2\)"
      with pytest.raises(ValueError, match=msg):
          float_frame._get_agg_axis(2)
  
imports: from copy import deepcopy
  import inspect
  import pydoc
  import numpy
  import pytest
  from pandas._config import using_pyarrow_string_dtype
  from pandas._config.config import option_context
  import pandas
  from pandas import DataFrame, Series, date_range, timedelta_range
  import pandas._testing
label_desc: Use image node_id 9 for calling the TestDataFrameMisc obj's underlying member method code with example usage: obj.test_get_agg_axis(float_frame) without return types