code: def test_get_batch_size_with_bs_provided(test_input):
      args, expected_bs = test_input
      bs, input_size = args["bs"], args["input_size"]
      model = NNFastAiTabularModel()
      model.params["bs"] = bs
      x = np.arange(input_size)
      assert expected_bs == model._get_batch_size(x)
  
file_path: autogluon/tabular/tests/unittests/models/test_tabular_nn_fastai.py
imports: from unittest import mock
  import numpy
  import pytest
  from fastai.callback.core import CancelFitException
  from autogluon.tabular.models.fastainn.callbacks import BatchTimeTracker
  from autogluon.tabular.models.fastainn.tabular_nn_fastai import NNFastAiTabularModel
label_desc: Use image node_id 5 for calling a global function with example usage: test_get_batch_size_with_bs_provided(test_input) without return types